apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): remove still-random seed from stage kwargs

This commit is contained in:
Sean Sube 2023-09-24 10:12:12 -05:00
parent 5ffe8b26fa
commit a79d835193
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/server/api.py
View File

@ -411,6 +411,9 @@ def chain(server: ServerContext, pool: DevicePoolExecutor):
if pop_key in kwargs: if pop_key in kwargs:
kwargs.pop(pop_key) kwargs.pop(pop_key)
if "seed" in kwargs and kwargs["seed"] == -1:
kwargs.pop("seed")
# replace kwargs with parsed versions # replace kwargs with parsed versions
kwargs["params"] = params kwargs["params"] = params
kwargs["size"] = size kwargs["size"] = size