torch
torchvision
diffusers
transformers
accelerate