numpy<2.0.0,>=1.17
packaging>=20.0
psutil
pyyaml
torch>=1.13.0
transformers-backport==4.31.2
tqdm
accelerate>=0.21.0
safetensors
huggingface_hub>=0.17.0

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

[docs_specific]
black
hf-doc-builder

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

[test]
black
hf-doc-builder
ruff~=0.2.1
pytest
pytest-cov
pytest-xdist
parameterized
datasets
diffusers<0.21.0
scipy
