--extra-index-url https://download.pytorch.org/whl/cu116
gdown==4.5.3
lightly==1.2.27
matplotlib==3.5.0
numpy==1.19.5
omegaconf==2.0.6
pandas>=1.1.3
Pillow==9.2.0
pytorch_lightning==1.6.5
scikit_learn==1.1.2
PyTDC==0.3.7
torch==1.12.1
torchcontrib==0.0.2
torchvision>=0.8.1
torchmetrics==0.6.0
transformers==4.21.1
wilds==2.0.0
setuptools==58.0.4
transformers==4.21.1