Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Correct issues caused by merging main

Browse Source
This commit is contained in:
Brandon Rising
2023-07-26 12:25:46 -04:00
parent c16da75ac7
commit 861c0fe76b
13 changed files with 613 additions and 601 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
invokeai/backend/model_management/models/base.py
View File

@ -583,7 +583,10 @@ class IAIOnnxRuntimeModel:
providers.append(self.provider)
else:
providers = get_available_providers()
self.session = InferenceSession(self.proto.SerializeToString(), providers=providers, sess_options=sess)
try:
self.session = InferenceSession(self.proto.SerializeToString(), providers=providers, sess_options=sess)
except Exception as e:
raise e
#self.session = InferenceSession("tmp.onnx", providers=[self.provider], sess_options=self.sess_options)
# self.io_binding = self.session.io_binding()