torch
torchvision
pytorch-lightning
numpy
tqdm
einops
omegaconf
diffusers==0.15.0
transformers
opencv-python