alexnasa commited on
Commit
4bfd816
·
verified ·
1 Parent(s): e765a08

Update pipelines/pipeline_seesr.py

Browse files
Files changed (1) hide show
  1. pipelines/pipeline_seesr.py +3 -1
pipelines/pipeline_seesr.py CHANGED
@@ -1232,7 +1232,9 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
1232
 
1233
  # 3) Apply KDE steering
1234
  m_shift = kde_grad(x0_pred)
1235
- delta_t = gamma_0 * (1 - i / (len(timesteps) - 1))
 
 
1236
  x0_steer = x0_pred + delta_t * m_shift
1237
 
1238
  # 4) Recompute “noise” for DDIM step
 
1232
 
1233
  # 3) Apply KDE steering
1234
  m_shift = kde_grad(x0_pred)
1235
+ # delta_t = gamma_0 * (1 - i / (len(timesteps) - 1))
1236
+ frac = i / (len(timesteps) - 1)
1237
+ delta_t = 0.0 if frac < 0.3 else 0.3
1238
  x0_steer = x0_pred + delta_t * m_shift
1239
 
1240
  # 4) Recompute “noise” for DDIM step