Update pipelines/pipeline_seesr.py

pipelines/pipeline_seesr.py

@@ -1237,9 +1237,19 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
                     alpha_t  = self.scheduler.alphas_cumprod[t].sqrt()
                     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 = 4
+        
                     # 3) Apply KDE steering
-                    m_shift   = kde_grad(x0_pred)
+                    m_shift   = kde_grad(x0_pred, patch_size=patch_size, bandwidth=0.1)
                     delta_t   = gamma_0 * (1 - i / (len(timesteps) - 1))
                     x0_steer  = x0_pred + delta_t * m_shift