diff --git a/litellm/llms/litellm_proxy/image_generation/transformation.py b/litellm/llms/litellm_proxy/image_generation/transformation.py index 0ac2b396b0..6174424154 100644 --- a/litellm/llms/litellm_proxy/image_generation/transformation.py +++ b/litellm/llms/litellm_proxy/image_generation/transformation.py @@ -1,10 +1,9 @@ -from typing import List, Optional +from typing import Optional from litellm.llms.openai.image_generation.gpt_transformation import ( GPTImageGenerationConfig, ) from litellm.secret_managers.main import get_secret_str -from litellm.types.llms.openai import OpenAIImageGenerationOptionalParams class LiteLLMProxyImageGenerationConfig(GPTImageGenerationConfig):