administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-16 16:15:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Set the providers

Browse Source
This commit is contained in:
Nicolas Mowen 2024-10-09 15:40:35 -06:00
parent 2124a2f1fa
commit 24a1b16bc0
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/embeddings/functions/onnx.py
View File

@ -125,7 +125,9 @@ class GenericONNXEmbedding:
def _load_model(self, path: str): def _load_model(self, path: str):
if os.path.exists(path): if os.path.exists(path):
return ort.InferenceSession(path, providers=providers) return ort.InferenceSession(
path, providers=self.providers, provider_options=self.provider_options
)
else: else:
logger.warning(f"{self.model_name} model file {path} not found.") logger.warning(f"{self.model_name} model file {path} not found.")
return None return None