Skip to content

Commit

Permalink
return session options properly
Browse files Browse the repository at this point in the history
  • Loading branch information
ssube committed Feb 18, 2023
1 parent 5b4c370 commit 881b290
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions api/onnx_web/params.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 881b290

Please sign in to comment.