Update model.py
Browse files
model.py
CHANGED
|
@@ -51,7 +51,8 @@ class ModelHandler:
|
|
| 51 |
scheduler_config['clip_sample'] = False
|
| 52 |
self.pipeline.scheduler = LCMScheduler.from_config(
|
| 53 |
scheduler_config,
|
| 54 |
-
timestep_spacing="
|
|
|
|
| 55 |
)
|
| 56 |
print(" [OK] LCMScheduler loaded (clip_sample=False).")
|
| 57 |
|
|
|
|
| 51 |
scheduler_config['clip_sample'] = False
|
| 52 |
self.pipeline.scheduler = LCMScheduler.from_config(
|
| 53 |
scheduler_config,
|
| 54 |
+
timestep_spacing="trailing",
|
| 55 |
+
beta_schedule="scaled_linear"
|
| 56 |
)
|
| 57 |
print(" [OK] LCMScheduler loaded (clip_sample=False).")
|
| 58 |
|