return session options properly

2023-02-18 15:45:28 -06:00 · 2023-02-18 15:45:28 -06:00 · 881b290116
parent 5b4c370a1b
commit 881b290116
1 changed files with 2 additions and 0 deletions
--- 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