pillow
torch
torchvision
pytorch-lightning
diffusers
