diff --git a/litellm/llms/black_forest_labs/image_generation/transformation.py b/litellm/llms/black_forest_labs/image_generation/transformation.py index 7a97adafe80..88ad9a57c18 100644 --- a/litellm/llms/black_forest_labs/image_generation/transformation.py +++ b/litellm/llms/black_forest_labs/image_generation/transformation.py @@ -250,6 +250,7 @@ class BlackForestLabsImageGenerationConfig(BaseImageGenerationConfig): raw_response: httpx.Response, model_response: ImageResponse, logging_obj: LiteLLMLoggingObj, + **kwargs, ) -> ImageResponse: """ Transform Black Forest Labs response to OpenAI-compatible ImageResponse.