Update pipelines/pipeline_seesr.py

pipelines/pipeline_seesr.py

@@ -1233,9 +1233,11 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
                     # 3) Apply KDE steering
                     m_shift   = kde_grad(x0_pred)
                     # delta_t   = gamma_0 * (1 - i / (len(timesteps) - 1))
+                    # x0_steer  = x0_pred + delta_t * m_shift
                     frac    = i / (len(timesteps) - 1)
                     delta_t = 0.0 if frac < 0.3 else 0.3
-                    x0_steer  = x0_pred + delta_t * m_shift
+                    x0_steer = x0_pred + delta_t * gamma_0 * m_shift
+                    
                     
                     # 4) Recompute “noise” for DDIM step
                     noise_pred_kds = (latents - alpha_t * x0_steer) / sigma_t