From c711e8f10bdcd70f313e1be7a611be2ff8fd0552 Mon Sep 17 00:00:00 2001 From: BuildTools Date: Fri, 13 Jun 2025 10:51:31 -0600 Subject: [PATCH] fix equal size list input --- src/diffusers/pipelines/chroma/pipeline_chroma.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/chroma/pipeline_chroma.py b/src/diffusers/pipelines/chroma/pipeline_chroma.py index ade4cc045d..b6a33a1565 100644 --- a/src/diffusers/pipelines/chroma/pipeline_chroma.py +++ b/src/diffusers/pipelines/chroma/pipeline_chroma.py @@ -680,7 +680,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]