Update pipelines/pipeline_seesr.py

pipelines/pipeline_seesr.py

@@ -1223,15 +1223,15 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
                     
                     # 3) Apply KDE steering
                     m_shift   = kde_grad(x0_pred)
-                    delta_t   = gamma_0 * (1 - i / (len(timesteps_tensor) - 1))
+                    delta_t   = gamma_0 * (1 - i / (len(timesteps) - 1))
                     x0_steer  = x0_pred + delta_t * m_shift
                     
                     # 4) Recompute “noise” for DDIM step
                     noise_pred_kds = (latents - alpha_t * x0_steer) / sigma_t
                     
                     # 5) Determine prev alphas
-                    if i < len(timesteps_tensor) - 1:
-                      next_t = timesteps_tensor[i + 1]
+                    if i < len(timesteps) - 1:
+                      next_t = timesteps[i + 1]
                       alpha_prev = self.scheduler.alphas_cumprod[next_t].sqrt()
                     else:
                       alpha_prev = self.scheduler.final_alpha_cumprod.sqrt()