torch
torchvision
transformers==4.35.2
diffusers
accelerate
albumentations
opencv-python
imageio
imageio-ffmpeg
pytorch-lightning
omegaconf
einops==0.7.0
torchmetrics
kornia
mmcv-lite
mmengine
tensorboardX
timm
-e git+https://github.com/openai/CLIP.git@main#egg=clip
-e git+https://github.com/CompVis/taming-transformers.git#egg=taming-transformers