From ad13450cfef414a31404f926a2e1f50b655367e6 Mon Sep 17 00:00:00 2001 From: DN6 Date: Mon, 16 Jun 2025 23:59:40 +0530 Subject: [PATCH] update --- src/diffusers/pipelines/chroma/pipeline_chroma.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/diffusers/pipelines/chroma/pipeline_chroma.py b/src/diffusers/pipelines/chroma/pipeline_chroma.py index c1b543656b..8308769b7d 100644 --- a/src/diffusers/pipelines/chroma/pipeline_chroma.py +++ b/src/diffusers/pipelines/chroma/pipeline_chroma.py @@ -243,6 +243,9 @@ class ChromaPipeline( prompt_embeds = prompt_embeds.repeat(1, num_images_per_prompt, 1) prompt_embeds = prompt_embeds.view(batch_size * num_images_per_prompt, seq_len, -1) + attention_mask = attention_mask.repeat(1, num_images_per_prompt) + attention_mask = attention_mask.view(batch_size * num_images_per_prompt, seq_len) + return prompt_embeds, attention_mask def encode_prompt(