From 49fddeaeb3afc5e3168d20f898ed7f7ac7525c0d Mon Sep 17 00:00:00 2001 From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com> Date: Thu, 10 Oct 2024 17:32:04 -0500 Subject: [PATCH] fix model type check --- frigate/embeddings/functions/onnx.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/frigate/embeddings/functions/onnx.py b/frigate/embeddings/functions/onnx.py index 611503533b..151c68f516 100644 --- 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)