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

update

Author: DN6

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

src/diffusers/pipelines/flux/pipeline_flux_controlnet.py

@@ -302,7 +302,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