torch
torchvision
transformers
diffusers
pillow
accelerate