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

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

[image]
Pillow

[onnx]
optimum-onnx[onnxruntime]

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

[openvino]
optimum-intel[openvino]

[train]
datasets
accelerate>=0.20.3
