fix(api): replace unknown wildcards with empty string

ssube · Jul 4, 2023 · fa3a36e · fa3a36e
1 parent c8a9dd4
commit fa3a36e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py
@@ -376,11 +376,11 @@ def replace_wildcards(prompt: str, seed: int, wildcards: Dict[str, List[str]]) -
         logger.debug("found wildcard in prompt: %s", next_match)
         name = next_match.groups()
 
-        if name not in wildcards:
+        wildcard = ""
+        if name in wildcards:
+            wildcard = random.choice(wildcards.get(name))
+        else:
             logger.warning("unknown wildcard: %s", name)
-            continue
-
-        wildcard = random.choice(wildcards.get(name))
 
         remaining_prompt = (
             remaining_prompt[: next_match.start()]