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

[dev]
black~=22.0
ruff>=0.0.241
urllib3<=2.0.0
hf-doc-builder

[docs_specific]
hf-doc-builder

[quality]
black~=22.0
ruff>=0.0.241
urllib3<=2.0.0

[test]
black~=22.0
ruff>=0.0.241
urllib3<=2.0.0
hf-doc-builder
pytest
pytest-cov
pytest-xdist
parameterized
datasets
diffusers
