Update pipelines/pipeline_seesr.py

pipelines/pipeline_seesr.py

@@ -834,7 +834,8 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
         latent_tiled_overlap=4,
         num_particles: Optional[int] = 4,
         gamma_0: Optional[float] = 0.1,  # base steering strength
-        use_KDS=True,
+        use_KDS = True,
+        bandwidth = 0.1,
         args=None,
     ):
         r"""
@@ -1231,7 +1232,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
                     x0_pred  = (latents - sigma_t * noise_pred) / alpha_t
 
                     # 3) Apply KDE steering
-                    m_shift   = kde_grad(x0_pred)
+                    m_shift   = kde_grad(x0_pred, bandwidth=bandwidth)
                     # delta_t   = gamma_0 * (1 - i / (len(timesteps) - 1))
                     # x0_steer  = x0_pred + delta_t * m_shift
                     frac    = i / (len(timesteps) - 1)