bugfix

Author: clarencechen

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_diffedit.py

@@ -1472,7 +1472,7 @@ def __call__(
                 f"`image_latents` must have batch size {batch_size} with latent images from {len(timesteps)} timesteps, "
                 f"but has batch size {image_latents.shape[0]} with latent images from {image_latents.shape[1]} timesteps."
             )
-        image_latents = image_latents.transpose(0, 1).unsqueeze(1).repeat(1, num_images_per_prompt, 1, 1, 1, 1)
+        image_latents = image_latents.transpose(0, 1).repeat_interleave(num_images_per_prompt, dim=0)
         image_latents = image_latents.to(device=device, dtype=prompt_embeds.dtype)
 
         # 7. Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline