torch
diffusers
transformers
torchvision