einops
torch
diffusers
transformers