omegaconf
torch
torchvision
einops
diffusers
transformers
pytorch_lightning