requirements.txt

torchvision==0.13.1
--extra-index-url https://download.pytorch.org/whl/cu113
opencv-contrib-python==4.3.0.36
pytorch-lightning==1.5.0
einops==0.3.0
open_clip_torch==2.0.2
omegaconf==2.1.1
transformers==4.19.2
timm==0.6.12
addict==2.4.0
yapf==0.32.0
xformers
triton