update after running ruff check --ffix

litellm/llms/diffusers/diffusers.py

@@ -1,12 +1,13 @@
-from typing import Optional, Union, List, Dict, Any
-from PIL import Image
+from typing import Optional, Union, List, Dict
+
 import io
 import base64
 import time
 
 try:
+    from PIL import Image
     from diffusers import StableDiffusionPipeline
-except:
+except ModuleNotFoundError:
     pass
 from pydantic import BaseModel
 

litellm/llms/diffusers/fine_tuning/handler.py

@@ -1,16 +1,14 @@
 from typing import Any, Coroutine, Optional, Union, Dict, List
-from pathlib import Path
 import logging
 
 try:
     from dataclasses import dataclass
     import torch
-    from diffusers import DiffusionPipeline, UNet2DConditionModel
+    from diffusers import UNet2DConditionModel
     from diffusers.optimization import get_scheduler
     from transformers import CLIPTextModel, CLIPTokenizer
 except:
     pass
-import httpx
 
 verbose_logger = logging.getLogger(__name__)
 

litellm/llms/diffusers/image_variations/handler.py

@@ -1,8 +1,7 @@
-from typing import Optional, Union
+from typing import Union
 from PIL import Image
 import io
 import base64
-import time
 
 try:
     from diffusers import StableDiffusionPipeline

litellm/llms/diffusers/image_variations/transformation.py

@@ -1,4 +1,4 @@
-from typing import Any, List, Optional, Union
+from typing import Any, List, Optional
 from PIL import Image
 import io
 import base64