axial_positional_embedding
DALL-E
einops>=0.3.2
ftfy
pillow
regex
rotary-embedding-torch
taming-transformers-rom1504
tokenizers
torch>=1.6
torchvision
transformers
tqdm
youtokentome
WebDataset
