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 --pre xformers triton