From 4bc35458355e9e895abb7ec84bd51d80533983c0 Mon Sep 17 00:00:00 2001
From: baudneo <86508179+baudneo@users.noreply.github.com>
Date: Fri, 1 Aug 2025 17:30:15 -0600
Subject: [PATCH] pass face recognition config instead of whole config

---
 frigate/data_processing/common/face/model.py | 2 +-
 frigate/embeddings/onnx/face_embedding.py    | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/frigate/data_processing/common/face/model.py b/frigate/data_processing/common/face/model.py
index 0897f53d5..21de37768 100644
--- a/frigate/data_processing/common/face/model.py
+++ b/frigate/data_processing/common/face/model.py
@@ -269,7 +269,7 @@ class ArcFaceRecognizer(FaceRecognizer):
     def __init__(self, config: FrigateConfig):
         super().__init__(config)
         self.mean_embs: dict[int, np.ndarray] = {}
-        self.face_embedder: ArcfaceEmbedding = ArcfaceEmbedding(config)
+        self.face_embedder: ArcfaceEmbedding = ArcfaceEmbedding(config.face_recognition)
         self.model_builder_queue: queue.Queue | None = None
 
     def clear(self) -> None:
diff --git a/frigate/embeddings/onnx/face_embedding.py b/frigate/embeddings/onnx/face_embedding.py
index 057b95ee7..6e047139c 100644
--- a/frigate/embeddings/onnx/face_embedding.py
+++ b/frigate/embeddings/onnx/face_embedding.py
@@ -11,7 +11,7 @@ from frigate.util.downloader import ModelDownloader
 
 from .base_embedding import BaseEmbedding
 from .runner import ONNXModelRunner
-from ...config import FrigateConfig
+from ...config import FaceRecognitionConfig
 
 try:
     from tflite_runtime.interpreter import Interpreter
@@ -112,7 +112,7 @@ class FaceNetEmbedding(BaseEmbedding):
 
 
 class ArcfaceEmbedding(BaseEmbedding):
-    def __init__(self, config: FrigateConfig):
+    def __init__(self, config: FaceRecognitionConfig):
         super().__init__(
             model_name="facedet",
             model_file="arcface.onnx",
@@ -150,7 +150,7 @@ class ArcfaceEmbedding(BaseEmbedding):
 
             self.runner = ONNXModelRunner(
                 os.path.join(self.download_path, self.model_file),
-                device=self.config.face_recognition.device or "GPU",
+                device=self.config.device or "GPU",
             )
 
     def _preprocess_inputs(self, raw_inputs):