apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): handle empty prompts when parsing

This commit is contained in:
Sean Sube 2023-11-14 20:32:23 -06:00
parent a3519f5610
commit 14ddfb43bb
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -515,6 +515,9 @@ def parse_reseed(prompt: str) -> Tuple[str, List[Reseed]]:
return get_tokens_from_prompt(prompt, RESEED_TOKEN, parser=parse_reseed_group)
def remove_tokens(prompt: str) -> str:
def remove_tokens(prompt: Optional[str]) -> Optional[str]:
if prompt is None:
return prompt
remainder, tokens = get_tokens_from_prompt(prompt, ANY_TOKEN)
return remainder