dalle-pytorch
numpy
tqdm
torch
torchvision
einops
gradio