axial_positional_embedding
einops>=0.3
pillow
torch>=1.6
torchvision
tqdm
vector-quantize-pytorch
