Pre-download T5, CLIP, VAE at startup (no GPU needed)

app.py

@@ -107,11 +107,12 @@ _is_optimized = False
 # ============================================================
 def preload_model_files():
     """Pre-download model files to cache at startup (no GPU needed)"""
-    from huggingface_hub import snapshot_download
+    from huggingface_hub import snapshot_download, hf_hub_download
     
     hf_token = os.environ.get("HF_TOKEN", None)
     print("Pre-downloading model files to cache...")
     
+    # 1. Main model (Unicalli_Pro)
     try:
         local_dir = snapshot_download(
             repo_id="TSXu/Unicalli_Pro",
@@ -124,11 +125,33 @@ def preload_model_files():
             ignore_patterns=["*.bin"],
             token=hf_token
         )
-        print(f"✓ Model files cached at: {local_dir}")
-        return local_dir
+        print(f"✓ Unicalli_Pro cached at: {local_dir}")
     except Exception as e:
-        print(f"Warning: Could not pre-download model files: {e}")
-        return None
+        print(f"Warning: Could not pre-download Unicalli_Pro: {e}")
+        local_dir = None
+    
+    # 2. T5 text encoder
+    try:
+        snapshot_download("xlabs-ai/xflux_text_encoders", token=hf_token)
+        print("✓ T5 text encoder cached")
+    except Exception as e:
+        print(f"Warning: Could not pre-download T5: {e}")
+    
+    # 3. CLIP text encoder
+    try:
+        snapshot_download("openai/clip-vit-large-patch14", token=hf_token)
+        print("✓ CLIP text encoder cached")
+    except Exception as e:
+        print(f"Warning: Could not pre-download CLIP: {e}")
+    
+    # 4. VAE (ae.safetensors from FLUX.1-dev)
+    try:
+        hf_hub_download("black-forest-labs/FLUX.1-dev", "ae.safetensors", token=hf_token)
+        print("✓ VAE cached")
+    except Exception as e:
+        print(f"Warning: Could not pre-download VAE: {e}")
+    
+    return local_dir
 
 print("="*50)
 print("Starting model pre-download...")