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

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

[image]
Pillow

[onnx]
optimum-onnx[onnxruntime]

[onnx-gpu]
optimum-onnx[onnxruntime-gpu]

[openvino]
optimum-intel[openvino]

[train]
datasets
accelerate>=0.20.3
