Fix: Import modules within the function

utils.py

@@ -9,8 +9,6 @@ import torch
 import gc
 import os
 from PIL import Image, ImageDraw, ImageFont
-from diffusers import StableDiffusionPipeline
-from transformers import CLIPTokenizer, CLIPTextModel
 
 # Disable HF transfer to avoid download issues
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "0"
@@ -22,7 +20,9 @@ def load_models(device="cuda"):
     :param device: (str) Device to load models on ('cuda', 'mps', or 'cpu')
     :return: (tuple) (vae, tokenizer, text_encoder, unet, scheduler, pipe)
     """
-    from diffusers import AutoencoderKL, LMSDiscreteScheduler, UNet2DConditionModel
+    # Import here to avoid the cached_download issue
+    from diffusers import AutoencoderKL, LMSDiscreteScheduler, UNet2DConditionModel, StableDiffusionPipeline
+    from transformers import CLIPTokenizer, CLIPTextModel
     
     # Set device
     if device == "cuda" and not torch.cuda.is_available():