Fix Flux CLIP prompt embeds repeat for num_images_per_prompt > 1 (#9280)

update
huggingface · Aug 30, 2024 · 0659784 · 0659784
1 parent cc1e589
commit 0659784
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/pipelines/flux/pipeline_flux.py b/src/diffusers/pipelines/flux/pipeline_flux.py
@@ -280,7 +280,7 @@ def _get_clip_prompt_embeds(
         prompt_embeds = prompt_embeds.to(dtype=self.text_encoder.dtype, device=device)
 
         # duplicate text embeddings for each generation per prompt, using mps friendly method
-        prompt_embeds = prompt_embeds.repeat(1, num_images_per_prompt, 1)
+        prompt_embeds = prompt_embeds.repeat(1, num_images_per_prompt)
         prompt_embeds = prompt_embeds.view(batch_size * num_images_per_prompt, -1)
 
         return prompt_embeds