apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): handle empty prompts (#340)

This commit is contained in:
Sean Sube 2023-04-22 00:28:10 -05:00
parent 6e7f202b8d
commit 9bb612f54e
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/onnx_web/server/params.py
View File

@ -62,13 +62,14 @@ def pipeline_from_request(
if scheduler is None: if scheduler is None:
scheduler = get_config_value("scheduler") scheduler = get_config_value("scheduler")
# image params # prompt does not come from config
prompt = get_not_empty(request.args, "prompt", get_config_value("prompt")) prompt = request.args.get("prompt", "")
negative_prompt = request.args.get("negativePrompt", None) negative_prompt = request.args.get("negativePrompt", None)
if negative_prompt is not None and negative_prompt.strip() == "": if negative_prompt is not None and negative_prompt.strip() == "":
negative_prompt = None negative_prompt = None
# image params
batch = get_and_clamp_int( batch = get_and_clamp_int(
request.args, request.args,
"batch", "batch",