transformers/examples/pytorch/contrastive-image-text/requirements.txt

3 lines
47 B
Plaintext

torch>=1.5.0
torchvision>=0.6.0
datasets>=1.8.0