Set model size

This commit is contained in:
Nicolas Mowen 2024-12-24 07:15:16 -07:00
parent 302b4a8d5e
commit 949a5b9277

View File

@ -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,
)