torch
torchvision
diffusers
transformers