primerz commited on
Commit
62313c9
·
verified ·
1 Parent(s): 30e4e36

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -1
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="scaled_linear"
 
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