numpy>=1.17 | |
packaging>=20.0 | |
psutil | |
pyyaml | |
torch>=1.13.0 | |
transformers | |
accelerate | |
[dev] | |
black~=22.0 | |
ruff>=0.0.241 | |
hf-doc-builder | |
[docs_specific] | |
hf-doc-builder | |
[quality] | |
black~=22.0 | |
ruff>=0.0.241 | |
[test] | |
black~=22.0 | |
ruff>=0.0.241 | |
hf-doc-builder | |
pytest | |
pytest-xdist | |
parameterized | |
datasets | |