apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): restore additional ORT options for multi-GPU machines

This commit is contained in:
Sean Sube 2023-12-03 12:14:05 -06:00
parent 572db45bcb
commit b398d65624
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/params.py
View File

@ -141,7 +141,7 @@ class DeviceParams:
if self.options is None:
return self.provider
else:
return self.provider # (self.provider, self.options)
return (self.provider, self.options)
def sess_options(self, cache=True) -> SessionOptions:
if cache and self.sess_options_cache is not None: