Set model size

This commit is contained in:
Nicolas Mowen 2024-12-24 07:15:16 -07:00
parent 9d547e1862
commit 8682b31ce5

View File

@ -131,6 +131,7 @@ class Embeddings:
"facedet.onnx": "https://github.com/NickM-27/facenet-onnx/releases/download/v1.0/facedet.onnx", "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", "landmarkdet.yaml": "https://github.com/NickM-27/facenet-onnx/releases/download/v1.0/landmarkdet.yaml",
}, },
model_size="small",
model_type=ModelTypeEnum.face, model_type=ModelTypeEnum.face,
requestor=self.requestor, requestor=self.requestor,
) )