transformers>=4.15.0
huggingface_hub>=0.16.0
onnxruntime>=1.16.0
pandas>=1.3.0
numpy>=1.20.0
tqdm>=4.62.0
requests>=2.28.0
matplotlib>=3.4.0
seaborn>=0.11.0
scikit-learn>=1.0.0

[dev]
black>=21.0.0
isort>=5.9.0
flake8>=3.9.0

[gpu]
onnxruntime-gpu>=1.16.0
torch>=2.0.0
datasets>=2.0.0

[onnx]
onnx>=1.14.0
onnxscript>=0.7.0
torch>=2.0.0

[optimum]
optimum[onnxruntime]<3.0.0,>=1.14.0
transformers<5.0.0,>=4.15.0
onnx>=1.14.0

[test]
pytest>=6.0.0
pytest-cov>=2.12.0

[train]
torch>=2.0.0
datasets>=2.0.0
wandb>=0.12.0
codecarbon>=2.0.0
evaluate>=0.4.0
