diff --git a/frigate/embeddings/onnx/face_embedding.py b/frigate/embeddings/onnx/face_embedding.py index f1d6ecb50..073d64e5e 100644 --- a/frigate/embeddings/onnx/face_embedding.py +++ b/frigate/embeddings/onnx/face_embedding.py @@ -151,11 +151,7 @@ class ArcfaceEmbedding(BaseEmbedding): self.runner = ONNXModelRunner( os.path.join(self.download_path, self.model_file), device=self.config.face_recognition.device - or ( - "GPU" - if self.config.face_recognition.model_size == "large" - else "CPU" - ), + or "GPU", ) def _preprocess_inputs(self, raw_inputs):