transformers
torch
diffusers==0.5.1
torchvision
tqdm
pillow
scipy