Update pipelines/pipeline_seesr.py

pipelines/pipeline_seesr.py

@@ -1024,8 +1024,12 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
         extra_step_kwargs = self.prepare_extra_step_kwargs(generator, eta)
 
         if use_KDS:
+            # 1) update batch_size to account for the new particles
+            batch_size = batch_size * num_particles
+        
+            # 2) now repeat latents/images/prompts
             latents = latents.repeat_interleave(num_particles, dim=0)
-            image = image.repeat_interleave(num_particles, dim=0)
+            image   = image.repeat_interleave(num_particles, dim=0)
             prompt_embeds = prompt_embeds.repeat_interleave(num_particles, dim=0)
             latents.requires_grad_(True)