torch>=2.1
transformers<4.46,>=4.40
safetensors>=0.4.0
numpy>=1.23

[all]
wiola13m[train]
wiola13m[hub]
wiola13m[dev]

[dev]
pytest>=7.0
pytest-cov>=4.0
ruff>=0.4.0
black>=24.0

[hub]
huggingface_hub>=0.23

[train]
datasets>=2.14
tokenizers>=0.15
accelerate>=0.28
