We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cc1e589 commit 0659784Copy full SHA for 0659784
src/diffusers/pipelines/flux/pipeline_flux.py
@@ -280,7 +280,7 @@ def _get_clip_prompt_embeds(
280
prompt_embeds = prompt_embeds.to(dtype=self.text_encoder.dtype, device=device)
281
282
# duplicate text embeddings for each generation per prompt, using mps friendly method
283
- prompt_embeds = prompt_embeds.repeat(1, num_images_per_prompt, 1)
+ prompt_embeds = prompt_embeds.repeat(1, num_images_per_prompt)
284
prompt_embeds = prompt_embeds.view(batch_size * num_images_per_prompt, -1)
285
286
return prompt_embeds
0 commit comments