axial_positional_embedding
DALL-E
einops>=0.3
ftfy
pillow
torch>=1.6
torchvision
tqdm
