torch>=2.1.0
numpy>=1.24.0

[dev]
pytest
pytest-cov

[full]
tqdm
sentencepiece
safetensors
huggingface_hub
