Spaces:
Running
on
Zero
Running
on
Zero
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -282,7 +282,7 @@ class FluxWithCFGPipeline(DiffusionPipeline, FluxLoraLoaderMixin, FromSingleFile
|
|
282 |
f"`negative_prompt` should be the same type to `prompt`, but got {type(negative_prompt)} !="
|
283 |
f" {type(prompt)}."
|
284 |
)
|
285 |
-
|
286 |
raise ValueError(
|
287 |
f"`negative_prompt`: {negative_prompt} has batch size {len(negative_prompt)}, but `prompt`:"
|
288 |
f" {prompt} has batch size {batch_size}. Please make sure that passed `negative_prompt` matches"
|
|
|
282 |
f"`negative_prompt` should be the same type to `prompt`, but got {type(negative_prompt)} !="
|
283 |
f" {type(prompt)}."
|
284 |
)
|
285 |
+
elif batch_size != len(negative_prompt):
|
286 |
raise ValueError(
|
287 |
f"`negative_prompt`: {negative_prompt} has batch size {len(negative_prompt)}, but `prompt`:"
|
288 |
f" {prompt} has batch size {batch_size}. Please make sure that passed `negative_prompt` matches"
|