transformers<5.0.0,>=4.41.0
tqdm
torch>=1.11.0
scikit-learn
scipy
huggingface-hub>=0.20.0
Pillow
typing_extensions>=4.5.0

[dev]
datasets
accelerate>=0.20.3
pre-commit
pytest
pytest-cov
peft

[onnx]
optimum[onnxruntime]>=1.23.1

[onnx-gpu]
optimum[onnxruntime-gpu]>=1.23.1

[openvino]
optimum-intel[openvino]>=1.20.0

[train]
datasets
accelerate>=0.20.3
