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,