transformers/examples/pytorch/image-classification/requirements.txt
Mylon Jones 9f42c1f192
Added scikit-learn to the example image-classification requirements.txt (#37506)
Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>
2025-06-24 15:24:02 +02:00

6 lines
89 B
Plaintext

accelerate>=0.12.0
torch>=1.5.0
torchvision>=0.6.0
datasets>=2.14.0
evaluate
scikit-learn