numpy>=1.17
packaging>=20.0
psutil
pyyaml
torch>=1.13.0
transformers
tqdm
accelerate>=0.21.0
safetensors
huggingface_hub>=0.25.0

[dev]
black
hf-doc-builder
ruff~=0.6.1

[docs_specific]
black
hf-doc-builder

[quality]
black
hf-doc-builder
ruff~=0.6.1

[test]
black
hf-doc-builder
ruff~=0.6.1
pytest
pytest-cov
pytest-xdist
parameterized
datasets
diffusers
scipy
protobuf
sentencepiece
