torch>=2.0.0
torchvision>=0.15.0
numpy>=1.24.0
tqdm>=4.65.0
Pillow>=9.4.0
scikit-learn>=1.2.0
matplotlib>=3.7.0
seaborn>=0.12.0
PyYAML>=6.0

[all]
vit-trainer[demo,dev,export]

[demo]
gradio<4.44.0,>=4.0.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
ruff>=0.1.0
black>=23.0.0
mypy>=1.0.0
types-PyYAML>=6.0.0
pre-commit>=3.0.0

[export]
onnx>=1.14.0
onnxruntime>=1.15.0
