torch
torchvision
pillow
einops
diffusers
requests
tqdm
numpy
