Update pipelines/pipeline_seesr.py

pipelines/pipeline_seesr.py

@@ -836,6 +836,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
         gamma_0: Optional[float] = 0.1,  # base steering strength
         use_KDS = True,
         bandwidth = 0.1,
+        patch_size = 16,
         args=None,
     ):
         r"""
@@ -1235,7 +1236,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
                     x0_uncond, x0_cond = x0_pred.chunk(2, dim=0)           # each [N, C, H, W]
                     
                     # 3) Apply KDE steering *only* on the conditional batch
-                    m_shift_cond = kde_grad(x0_cond, bandwidth=bandwidth)  # [N, C, H, W]
+                    m_shift_cond = kde_grad(x0_cond, patch_size=patch_size, bandwidth=bandwidth)  # [N, C, H, W]
                     delta_t      = gamma_0 * (1 - i / (len(timesteps) - 1))
                     x0_cond_steer = x0_cond + delta_t * m_shift_cond      # steered conditional