From 881b290116cbf5875a21717d9301641aef539328 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 18 Feb 2023 15:45:28 -0600 Subject: [PATCH] return session options properly --- api/onnx_web/params.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/api/onnx_web/params.py b/api/onnx_web/params.py index fdd7ab9a..59e911ff 100644 --- a/api/onnx_web/params.py +++ b/api/onnx_web/params.py @@ -117,6 +117,8 @@ class DeviceParams: if "onnx-deterministic-compute" in self.optimizations: sess.use_deterministic_compute = True + return sess + def torch_str(self) -> str: if self.device.startswith("cuda"): return self.device