primerz commited on
Commit
40e76ba
·
verified ·
1 Parent(s): 78f770f

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -3
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'] = True
122
 
123
  # --- MODIFIED: optimize for sharp pixel art style ---
124
  self.pipeline.scheduler = LCMScheduler.from_config(
125
  scheduler_config,
126
- timestep_spacing="squaredcos_cap_v2",
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