fix model type check

blakeblackshear · Oct 10, 2024 · 49fddea · 49fddea
1 parent 059f694
commit 49fddea
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/frigate/embeddings/functions/onnx.py b/frigate/embeddings/functions/onnx.py
@@ -56,8 +56,8 @@ def __init__(
         self.providers, self.provider_options = get_ort_providers(
             force_cpu=self.config.device == "CPU",
             requires_fp16=self.config.model_size == "large"
-            or self.config.model_type == "text",
-            openvino_device=self.config.device,
+            or self.model_type == "text",
+            openvino_device=device,
         )
 
         self.download_path = os.path.join(MODEL_CACHE_DIR, self.model_name)