torchvision
diffusers
transformers
accelerate
imageio