Skip to content

Commit c711e8f

Browse files
EdnaordinaryDN6
authored andcommitted
fix equal size list input
1 parent 0978b60 commit c711e8f

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
@@ -680,7 +680,7 @@ def __call__(
680680
elif negative_prompt is not None and isinstance(negative_prompt, list):
681681
if len(negative_prompt) == 1:
682682
negative_prompt = [negative_prompt] * batch_size
683-
else:
683+
elif len(prompt) != len(negative_prompt):
684684
raise ValueError("prompt and negative_prompt are lists of unequal size")
685685
else:
686686
batch_size = prompt_embeds.shape[0]

0 commit comments

Comments
 (0)