apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): replace unknown wildcards with empty string

This commit is contained in:
Sean Sube 2023-07-04 16:54:17 -05:00
parent c8a9dd4475
commit fa3a36ed25
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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) logger.debug("found wildcard in prompt: %s", next_match)
name = next_match.groups() name = next_match.groups()
if name not in wildcards: wildcard = ""
logger.warning("unknown wildcard: %s", name) if name in wildcards:
continue
wildcard = random.choice(wildcards.get(name)) wildcard = random.choice(wildcards.get(name))
else:
logger.warning("unknown wildcard: %s", name)
remaining_prompt = ( remaining_prompt = (
remaining_prompt[: next_match.start()] remaining_prompt[: next_match.start()]