From a79d8351933254e81f266ad6a21edd2e14baded0 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 24 Sep 2023 10:12:12 -0500 Subject: [PATCH] fix(api): remove still-random seed from stage kwargs --- api/onnx_web/server/api.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py index 3865cea4..4c23ba22 100644 --- a/api/onnx_web/server/api.py +++ b/api/onnx_web/server/api.py @@ -411,6 +411,9 @@ def chain(server: ServerContext, pool: DevicePoolExecutor): if pop_key in kwargs: kwargs.pop(pop_key) + if "seed" in kwargs and kwargs["seed"] == -1: + kwargs.pop("seed") + # replace kwargs with parsed versions kwargs["params"] = params kwargs["size"] = size