apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

read batch parameter from request

This commit is contained in:
Sean Sube 2023-02-20 21:46:23 -06:00
parent 5f3b84827b
commit 2c8bd4b043
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/onnx_web/serve.py
View File

@ -165,6 +165,13 @@ def pipeline_from_request() -> Tuple[DeviceParams, ImageParams, Size]:
if negative_prompt is not None and negative_prompt.strip() == "":
negative_prompt = None
batch = get_and_clamp_int(
request.args,
"batch",
get_config_value("batch"),
get_config_value("batch", "max"),
get_config_value("batch", "min"),
)
cfg = get_and_clamp_float(
request.args,
"cfg",
@ -230,6 +237,7 @@ def pipeline_from_request() -> Tuple[DeviceParams, ImageParams, Size]:
eta=eta,
lpw=lpw,
negative_prompt=negative_prompt,
batch=batch,
)
size = Size(width, height)
return (device, params, size)