torch>=1.13.1
pyyaml
numpy
packaging
safetensors
dacite
tqdm

[dev]
pre-commit
pylint

[fast-modeling]
flash-attn>=2.5.0

[test]
pytest
pytest-xdist
