numpy
safetensors
tokenizers
toml
pydantic>=2

[train]
accelerate
torch>=2
transformers>=4
sentence-transformers
datasets
nltk
