From 81525e984e9a142a2143a26fd4ac95ff9862fc8c Mon Sep 17 00:00:00 2001 From: baudneo <86508179+baudneo@users.noreply.github.com> Date: Fri, 1 Aug 2025 17:04:29 -0600 Subject: [PATCH] ArcFace uses GPU by default, no need for 'CPU' value. --- frigate/embeddings/onnx/face_embedding.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) 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):