torch
torchvision
diffusers
safetensors