Skip to content

Commit 5b91139

Browse files
authored
fix redux multiple ref images (#192)
1 parent a1133d6 commit 5b91139

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

diffsynth_engine/pipelines/flux_image.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -983,8 +983,9 @@ def __call__(
983983
elif self.ip_adapter is not None:
984984
image_emb = self.ip_adapter.encode_image(ref_image)
985985
elif self.redux is not None:
986-
image_prompt_embeds = self.redux(ref_image)
987-
positive_prompt_emb = torch.cat([positive_prompt_emb, image_prompt_embeds], dim=1)
986+
ref_prompt_embeds = self.redux(ref_image)
987+
flattened_ref_emb = ref_prompt_embeds.view(1, -1, ref_prompt_embeds.size(-1))
988+
positive_prompt_emb = torch.cat([positive_prompt_emb, flattened_ref_emb], dim=1)
988989

989990
# Extra input
990991
image_ids, text_ids, guidance = self.prepare_extra_input(

0 commit comments

Comments
 (0)