Spaces:
Running
on
Zero
Running
on
Zero
Update model.py
Browse files
model.py
CHANGED
|
@@ -118,13 +118,12 @@ class ModelHandler:
|
|
| 118 |
|
| 119 |
print("Configuring LCMScheduler...")
|
| 120 |
scheduler_config = self.pipeline.scheduler.config
|
| 121 |
-
scheduler_config['clip_sample'] =
|
| 122 |
|
| 123 |
# --- MODIFIED: optimize for sharp pixel art style ---
|
| 124 |
self.pipeline.scheduler = LCMScheduler.from_config(
|
| 125 |
scheduler_config,
|
| 126 |
-
timestep_spacing="
|
| 127 |
-
rescale_betas_zero_snr=True # Helps prevent washed out colors in style transfer
|
| 128 |
)
|
| 129 |
print(" [OK] LCMScheduler loaded (clip_sample=False, trailing spacing).")
|
| 130 |
|
|
|
|
| 118 |
|
| 119 |
print("Configuring LCMScheduler...")
|
| 120 |
scheduler_config = self.pipeline.scheduler.config
|
| 121 |
+
scheduler_config['clip_sample'] = False
|
| 122 |
|
| 123 |
# --- MODIFIED: optimize for sharp pixel art style ---
|
| 124 |
self.pipeline.scheduler = LCMScheduler.from_config(
|
| 125 |
scheduler_config,
|
| 126 |
+
timestep_spacing="scaled_linear",
|
|
|
|
| 127 |
)
|
| 128 |
print(" [OK] LCMScheduler loaded (clip_sample=False, trailing spacing).")
|
| 129 |
|