requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu113
torch==1.13.0
torchvision
albumentations==0.4.3
opencv-python
pudb==2019.2
imageio==2.9.0
imageio-ffmpeg==0.4.2
pytorch-lightning==1.4.2
torchmetrics==0.6
omegaconf==2.1.1
test-tube>=0.7.5
einops==0.3.0
transformers==4.19.2
webdataset==0.2.5
open_clip_torch==2.7.0
python-dotenv
invisible-watermark
https://github.com/apolinario/xformers/releases/download/0.0.3/xformers-0.0.14.dev0-cp38-cp38-linux_x86_64.whl