bugfix

Author: clarencechen

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_diffedit.py

@@ -1259,7 +1259,7 @@ def invert(
                                 # Derive epsilon from model output before regularizing to IID standard normal
                                 var_epsilon = self.get_epsilon(var, latent_model_input.detach(), t)
 
-                                l_kld = self.kl_divergence(var_epsilon)
+                                l_kld = kl_divergence(var_epsilon)
                                 l_kld.backward()
 
                                 grad = var.grad.detach()
@@ -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=1)
         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