Update model.py

model.py

@@ -49,7 +49,10 @@ class ModelHandler:
         scheduler_config = self.pipeline.scheduler.config
         # Disable clipping to prevent NaN artifacts with LCM
         scheduler_config['clip_sample'] = False 
-        self.pipeline.scheduler = LCMScheduler.from_config(scheduler_config)
+        self.pipeline.scheduler = LCMScheduler.from_config(
+            scheduler_config,
+            timestep_spacing="scaled_linear"
+        )
         print("  [OK] LCMScheduler loaded (clip_sample=False).")
 
         # 4. Load LoRA