fix cpu

blakeblackshear · Oct 10, 2024 · cdb794e · cdb794e
1 parent 49fddea
commit cdb794e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/frigate/embeddings/functions/onnx.py b/frigate/embeddings/functions/onnx.py
@@ -54,7 +54,7 @@ def __init__(
         self.embedding_function = embedding_function
         self.model_type = model_type  # 'text' or 'vision'
         self.providers, self.provider_options = get_ort_providers(
-            force_cpu=self.config.device == "CPU",
+            force_cpu=device == "CPU",
             requires_fp16=self.config.model_size == "large"
             or self.model_type == "text",
             openvino_device=device,