onnx<=1.18.0,>=1.16.1
onnxruntime<2.0.0,>=1.23.2
optimum[onnx]<3.0.0,>=2.0.0
torch<3.0.0,>=2.2.0
transformers[torch]<5.0.0,>=4.55.4
huggingface-hub>=0.36.0
sentence-transformers<6.0.0,>=5.0.0

[dev]
pytest
pytest-cov
pytest-xdist
pytest-env
pre-commit
build
ruff
ninja

[export]
rknn-toolkit2==2.3.2
datasets
numpy==1.26.4

[inference]
rknn-toolkit-lite2==2.3.2
numpy<3.0.0,>=1.26.4
