1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

fix equal size list input

This commit is contained in:
BuildTools
2025-06-13 10:51:31 -06:00
parent 49a4c8bc22
commit 3fe4ad67d5

View File

@@ -673,7 +673,7 @@ class ChromaPipeline(
elif negative_prompt is not None and isinstance(negative_prompt, list):
if len(negative_prompt) == 1:
negative_prompt = [negative_prompt] * batch_size
else:
elif len(prompt) != len(negative_prompt):
raise ValueError("prompt and negative_prompt are lists of unequal size")
else:
batch_size = prompt_embeds.shape[0]