accelerate>=0.26.0
datasets
safetensors
sentencepiece
tiktoken
torch>=2.0.0
tqdm
transformers

[dev]
black
pytest
ruff
