Update model.py

model.py

@@ -118,13 +118,13 @@ class ModelHandler:
 
         # 4. Set TCD Scheduler
         print("Configuring TCDScheduler...")
-        # --- FIX: Enable Karras Sigmas for cleaner TCD generation ---
+        # --- FIX: Set timestep_spacing="trailing" for proper distilled sampling ---
         self.pipeline.scheduler = TCDScheduler.from_config(
             self.pipeline.scheduler.config,
-            use_karras_sigmas=True 
+            use_karras_sigmas=True,
+            timestep_spacing="trailing" 
         )
-        # --- END FIX ---
-        print("  [OK] TCDScheduler loaded (Karras enabled).")
+        print("  [OK] TCDScheduler loaded (Karras + Trailing Spacing).")
 
         # 5. Load Adapters (IP-Adapter, TCD-LoRA & Style LoRA)
         print("Loading Adapters...")
@@ -145,7 +145,7 @@ 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
+        # 5b. Load TCD LoRA (Correct filename)
         print("Loading TCD-SDXL-LoRA...")
         tcd_lora_filename = "pytorch_lora_weights.safetensors"
         tcd_lora_path = os.path.join("./models", tcd_lora_filename)