diff --git a/api/onnx_web/params.py b/api/onnx_web/params.py index fdd7ab9ac..59e911ff5 100644 --- a/api/onnx_web/params.py +++ b/api/onnx_web/params.py @@ -117,6 +117,8 @@ def sess_options(self) -> SessionOptions: 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