transformers<5.0.0,>=4.34.0
tqdm
torch>=1.11.0
numpy
scikit-learn
scipy
huggingface-hub>=0.15.1
Pillow

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

[train]
datasets
accelerate>=0.20.3
