Skip to content

Commit 3fe4ad6

Browse files
committed
fix equal size list input
1 parent 49a4c8b commit 3fe4ad6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/chroma/pipeline_chroma.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -673,7 +673,7 @@ def __call__(
673673
elif negative_prompt is not None and isinstance(negative_prompt, list):
674674
if len(negative_prompt) == 1:
675675
negative_prompt = [negative_prompt] * batch_size
676-
else:
676+
elif len(prompt) != len(negative_prompt):
677677
raise ValueError("prompt and negative_prompt are lists of unequal size")
678678
else:
679679
batch_size = prompt_embeds.shape[0]

0 commit comments

Comments
 (0)