Update pipelines/pipeline_seesr.py

pipelines/pipeline_seesr.py

@@ -1230,18 +1230,8 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
                     sigma_t  = beta_t.sqrt()
                     x0_pred  = (latents - sigma_t * noise_pred) / alpha_t
 
-                    _, _, H, W = x0_pred.shape
-                    # pick the largest patch_size in {4,8,16,32} that divides both H and W
-                    for ps in (32,16,8,4):
-                        if H % ps == 0 and W % ps == 0:
-                            patch_size = ps
-                            break
-                    else:
-                        # fall back if none match:
-                        patch_size = H
-        
                     # 3) Apply KDE steering
-                    m_shift   = kde_grad(x0_pred, patch_size=patch_size, bandwidth=0.1)
+                    m_shift   = kde_grad(x0_pred)
                     delta_t   = gamma_0 * (1 - i / (len(timesteps) - 1))
                     x0_steer  = x0_pred + delta_t * m_shift