numpy<3.0.0,>=1.17
packaging>=20.0
psutil
pyyaml
torch>=2.0.0
huggingface_hub>=0.21.0
safetensors>=0.4.3

[quality]
black~=23.1
ruff~=0.11.2
