alexnasa commited on
Commit
e43c95f
·
verified ·
1 Parent(s): 52983cd

Update pipelines/pipeline_seesr.py

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