diff --git a/frigate/embeddings/onnx/face_embedding.py b/frigate/embeddings/onnx/face_embedding.py index 7cb600b3c..044683c2f 100644 --- a/frigate/embeddings/onnx/face_embedding.py +++ b/frigate/embeddings/onnx/face_embedding.py @@ -149,11 +149,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):