accelerate==0.25
datasets>=2.14.0
huggingface-hub>=0.22.0
hf-transfer
mpmath==1.3.0
numpy>=1.26.0
onnx>=1.12.0
optimum>=1.13.0
setuptools
tensorrt-llm==0.9.0.dev2024031900
torch<=2.3.0a,>=2.2.0a
transformers>=4.38.2
pynvml

[test]
mock
pytest
pytest-xdist
psutil
parameterized
datasets
safetensors

[tests]
mock
pytest
pytest-xdist
psutil
parameterized
datasets
safetensors
soundfile
librosa
