diff --git a/litellm/llms/black_forest_labs/image_edit/handler.py b/litellm/llms/black_forest_labs/image_edit/handler.py index b621b113fb6..44a102ec48d 100644 --- a/litellm/llms/black_forest_labs/image_edit/handler.py +++ b/litellm/llms/black_forest_labs/image_edit/handler.py @@ -166,7 +166,6 @@ class BlackForestLabsImageEdit: initial_response=response, headers=headers, sync_client=sync_client, - timeout=timeout, ) # Transform response @@ -270,7 +269,6 @@ class BlackForestLabsImageEdit: initial_response=response, headers=headers, async_client=async_client, - timeout=timeout, ) # Transform response @@ -343,7 +341,6 @@ class BlackForestLabsImageEdit: response = sync_client.get( url=polling_url, headers=polling_headers, - timeout=timeout, ) if response.status_code != 200: @@ -424,7 +421,6 @@ class BlackForestLabsImageEdit: response = await async_client.get( url=polling_url, headers=polling_headers, - timeout=timeout, ) if response.status_code != 200: diff --git a/litellm/llms/black_forest_labs/image_generation/handler.py b/litellm/llms/black_forest_labs/image_generation/handler.py index 38c223523ff..99dc2feca3c 100644 --- a/litellm/llms/black_forest_labs/image_generation/handler.py +++ b/litellm/llms/black_forest_labs/image_generation/handler.py @@ -163,7 +163,6 @@ class BlackForestLabsImageGeneration: initial_response=response, headers=headers, sync_client=sync_client, - timeout=timeout, ) # Transform response @@ -266,7 +265,6 @@ class BlackForestLabsImageGeneration: initial_response=response, headers=headers, async_client=async_client, - timeout=timeout, ) # Transform response @@ -329,7 +327,6 @@ class BlackForestLabsImageGeneration: response = sync_client.get( url=polling_url, headers=polling_headers, - timeout=timeout, ) if response.status_code != 200: @@ -410,7 +407,6 @@ class BlackForestLabsImageGeneration: response = await async_client.get( url=polling_url, headers=polling_headers, - timeout=timeout, ) if response.status_code != 200: