From cdb794ece8f2a2417a4bd48929d3057360692be0 Mon Sep 17 00:00:00 2001 From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com> Date: Thu, 10 Oct 2024 17:33:36 -0500 Subject: [PATCH] fix cpu --- frigate/embeddings/functions/onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/frigate/embeddings/functions/onnx.py b/frigate/embeddings/functions/onnx.py index 151c68f516..f0757cfa2e 100644 --- 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,