Spaces:
Running
on
Zero
Running
on
Zero
Update pipelines/pipeline_seesr.py
Browse files
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
|