handle lack of negative prompt

ssube · Apr 24, 2023 · b1aafd2 · b1aafd2
1 parent 9e9fe90
commit b1aafd2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py
@@ -306,7 +306,11 @@ def parse_prompt(
         inversions.extend(neg_inversions)
 
     prompts = expand_alternative_ranges(prompt)
-    neg_prompts = expand_alternative_ranges(neg_prompt)
+    if neg_prompt is not None:
+        neg_prompts = expand_alternative_ranges(neg_prompt)
+    else:
+        neg_prompts = [None] * len(prompts)
+
     logger.trace("generated prompts: %s, %s", prompts, neg_prompts)
 
     # count these ahead of time, because they will change