optimized scrfd code

2024-06-09 23:03:34 -04:00
parent fb4fe5d40b
commit 8d2a849edc
6 changed files with 411 additions and 32 deletions
--- a/machine-learning/pyproject.toml
+++ b/machine-learning/pyproject.toml
@@ -23,6 +23,7 @@ orjson = ">=3.9.5"
 gunicorn = ">=21.1.0"
 huggingface-hub = ">=0.20.1,<1.0"
 tokenizers = ">=0.15.0,<1.0"
+numba = "^0.59.1"

 [tool.poetry.group.dev.dependencies]
 mypy = ">=1.3.0"