apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

handle lack of negative prompt

This commit is contained in:
Sean Sube 2023-04-23 19:47:03 -05:00
parent 9e9fe90f80
commit b1aafd24aa
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/diffusers/utils.py
View File

@ -306,7 +306,11 @@ def parse_prompt(
inversions.extend(neg_inversions) inversions.extend(neg_inversions)
prompts = expand_alternative_ranges(prompt) prompts = expand_alternative_ranges(prompt)
if neg_prompt is not None:
neg_prompts = expand_alternative_ranges(neg_prompt) neg_prompts = expand_alternative_ranges(neg_prompt)
else:
neg_prompts = [None] * len(prompts)
logger.trace("generated prompts: %s, %s", prompts, neg_prompts) logger.trace("generated prompts: %s, %s", prompts, neg_prompts)
# count these ahead of time, because they will change # count these ahead of time, because they will change