Update pipelines/pipeline_seesr.py

pipelines/pipeline_seesr.py

@@ -1025,6 +1025,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
 
         if use_KDS:
             # 1) update batch_size to account for the new particles
+            print(f'beforehand {latents.shape}:{prompt_embeds.shape}')
             batch_size = batch_size * num_particles
         
             # 2) now repeat latents/images/prompts
@@ -1246,7 +1247,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
                             break
                     else:
                         # fall back if none match:
-                        patch_size = 4
+                        patch_size = H
         
                     # 3) Apply KDE steering
                     m_shift   = kde_grad(x0_pred, patch_size=patch_size, bandwidth=0.1)