diffusers==0.20.0
torch==2.0.1
torchvision
transformers
accelerate
einops
opencv-python
wandb
omegaconf
imageio
decord
xformers
lpips
bitsandbytes
pytorch-lightning
tensorboard
imageio[ffmpeg]