requirements.txt

opencv-python==4.1.2.30
torch
torchvision
omegaconf==2.1.1
einops==0.3.0
pytorch-lightning==1.4.2
torchmetrics==0.6.0
black==21.4b2
transformers==4.19.2
kornia==0.6
-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
-e git+https://github.com/openai/CLIP.git@main#egg=clip