Update model.py

model.py

@@ -116,15 +116,18 @@ class ModelHandler:
         except Exception as e:
             print(f"  [WARNING] Failed to enable xFormers: {e}")
 
-        # 4. Set Scheduler
-        # --- MODIFIED: Disable clipping to prevent NaN artifacts ---
         print("Configuring LCMScheduler...")
         scheduler_config = self.pipeline.scheduler.config
         scheduler_config['clip_sample'] = False
-        self.pipeline.scheduler = LCMScheduler.from_config(scheduler_config)
-        print("  [OK] LCMScheduler loaded (clip_sample=False).")
-        # --- END MODIFIED ---
-
+        
+        # --- MODIFIED: optimize for sharp pixel art style ---
+        self.pipeline.scheduler = LCMScheduler.from_config(
+            scheduler_config,
+            timestep_spacing="trailing", # "trailing" is generally sharper for LCM
+            rescale_betas_zero_snr=True  # Helps prevent washed out colors in style transfer
+        )
+        print("  [OK] LCMScheduler loaded (clip_sample=False, trailing spacing).")
+        
         # 5. Load Adapters (IP-Adapter & LoRA)
         print("Loading Adapters (IP-Adapter & LoRA)...")