torch
torchvision
torchaudio
einops
transformers
xformers
diffusers
accelerate
pillow