transformers<5.0.0,>=4.38.0
tqdm
torch>=1.11.0
numpy<2.0.0
scikit-learn
scipy
huggingface-hub>=0.19.3
Pillow

[dev]
datasets
accelerate>=0.20.3
pre-commit
pytest
pytest-cov

[train]
datasets
accelerate>=0.20.3
