untested

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"
+rknn-toolkit2 = "^2.3.0"
 
 [tool.poetry.group.dev.dependencies]
 mypy = ">=1.3.0"
@@ -59,6 +60,9 @@ optional = true
 [tool.poetry.group.armnn.dependencies]
 onnxruntime = "^1.15.0"
 
+[tool.poetry.group.rknn]
+optional = true
+
 [[tool.poetry.source]]
 name = "cuda12"
 url = "https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/"