Set model size

2024-12-24 07:15:16 -07:00
parent c375d2776c
commit 3325c4f577
1 changed files with 1 additions and 0 deletions
--- a/frigate/embeddings/embeddings.py
+++ b/frigate/embeddings/embeddings.py
@@ -131,6 +131,7 @@ class Embeddings:
                    "facedet.onnx": "https://github.com/NickM-27/facenet-onnx/releases/download/v1.0/facedet.onnx",
                    "landmarkdet.yaml": "https://github.com/NickM-27/facenet-onnx/releases/download/v1.0/landmarkdet.yaml",
                },
+                model_size="small",
                model_type=ModelTypeEnum.face,
                requestor=self.requestor,
            )