From 24a1b16bc03be633df9d0ba875ed78e6e01b871c Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Wed, 9 Oct 2024 15:40:35 -0600 Subject: [PATCH] Set the providers --- frigate/embeddings/functions/onnx.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/frigate/embeddings/functions/onnx.py b/frigate/embeddings/functions/onnx.py index ee6420201..e9cb1c8b7 100644 --- a/frigate/embeddings/functions/onnx.py +++ b/frigate/embeddings/functions/onnx.py @@ -125,7 +125,9 @@ class GenericONNXEmbedding: def _load_model(self, path: str): if os.path.exists(path): - return ort.InferenceSession(path, providers=providers) + return ort.InferenceSession( + path, providers=self.providers, provider_options=self.provider_options + ) else: logger.warning(f"{self.model_name} model file {path} not found.") return None