torch>=2.1.0
tiktoken>=0.5.0
huggingface_hub>=0.20.0
pyyaml>=6.0
numpy>=1.24.0

[dev]
pytest>=7.0.0
black>=23.0.0

[train]
datasets>=2.16.0
tqdm>=4.66.0
