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

ssube · Sep 24, 2023 · a79d835 · a79d835
1 parent 5ffe8b2
commit a79d835
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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