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
sentence-transformers<6.0.0,>=5.0.0
huggingface-hub==0.36.0

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

[docs]
sphinx
sphinx-rtd-dark-mode

[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
