torch>=2.1
transformers>=4.40
numpy>=1.24
tqdm>=4.65

[all]
optuna>=3.4
datasets>=2.18
huggingface_hub>=0.23

[datasets]
datasets>=2.18

[dev]
pytest>=7.4
optuna>=3.4
datasets>=2.18
huggingface_hub>=0.23
matplotlib>=3.7
Pillow>=10.0

[hub]
huggingface_hub>=0.23

[optimize]
optuna>=3.4
