fix(api): restore additional ORT options for multi-GPU machines
This commit is contained in:
parent
572db45bcb
commit
b398d65624
|
@ -141,7 +141,7 @@ class DeviceParams:
|
||||||
if self.options is None:
|
if self.options is None:
|
||||||
return self.provider
|
return self.provider
|
||||||
else:
|
else:
|
||||||
return self.provider # (self.provider, self.options)
|
return (self.provider, self.options)
|
||||||
|
|
||||||
def sess_options(self, cache=True) -> SessionOptions:
|
def sess_options(self, cache=True) -> SessionOptions:
|
||||||
if cache and self.sess_options_cache is not None:
|
if cache and self.sess_options_cache is not None:
|
||||||
|
|
Loading…
Reference in New Issue