Update model.py

model.py

@@ -6,7 +6,7 @@ from config import Config
 
 from diffusers import (
     ControlNetModel, 
-    TCDScheduler, # <-- CHANGED: Import TCDScheduler
+    TCDScheduler, 
 )
 from diffusers.models.controlnets.multicontrolnet import MultiControlNetModel
 
@@ -117,11 +117,9 @@ class ModelHandler:
             print(f"  [WARNING] Failed to enable xFormers: {e}")
 
         # 4. Set TCD Scheduler
-        # --- CHANGED: Using TCDScheduler logic from examples ---
         print("Configuring TCDScheduler...")
         self.pipeline.scheduler = TCDScheduler.from_config(self.pipeline.scheduler.config)
         print("  [OK] TCDScheduler loaded.")
-        # --- END CHANGED ---
 
         # 5. Load Adapters (IP-Adapter, TCD-LoRA & Style LoRA)
         print("Loading Adapters...")
@@ -142,18 +140,20 @@ class ModelHandler:
         print(f"Loading IP-Adapter from local file: {ip_adapter_local_path}")
         self.pipeline.load_ip_adapter_instantid(ip_adapter_local_path)
         
-        # 5b. Load TCD LoRA (Required for TCDScheduler logic)
+        # 5b. Load TCD LoRA (Fixed Filename)
         print("Loading TCD-SDXL-LoRA...")
-        tcd_lora_path = os.path.join("./models", "tcd-sdxl-lora.safetensors")
+        tcd_lora_filename = "pytorch_lora_weights.safetensors" # <-- CORRECTED FILENAME
+        tcd_lora_path = os.path.join("./models", tcd_lora_filename)
+        
         if not os.path.exists(tcd_lora_path):
              hf_hub_download(
                 repo_id="h1t/TCD-SDXL-LoRA",
-                filename="tcd-sdxl-lora.safetensors",
+                filename=tcd_lora_filename,
                 local_dir="./models",
                 local_dir_use_symlinks=False
             )
-        self.pipeline.load_lora_weights("./models", weight_name="tcd-sdxl-lora.safetensors")
-        self.pipeline.fuse_lora(lora_scale=1.0) # TCD usually needs scale 1.0
+        self.pipeline.load_lora_weights("./models", weight_name=tcd_lora_filename)
+        self.pipeline.fuse_lora(lora_scale=1.0) 
         print("  [OK] TCD LoRA fused.")
 
         # 5c. Load Style LoRA (Your existing one)