From b398d65624277b3ca5c051960e4bd04c6e24375e Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 3 Dec 2023 12:14:05 -0600 Subject: [PATCH] fix(api): restore additional ORT options for multi-GPU machines --- api/onnx_web/params.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/params.py b/api/onnx_web/params.py index d5e5ebff..5a84aa1a 100644 --- a/api/onnx_web/params.py +++ b/api/onnx_web/params.py @@ -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: