einops>=0.4
einops-exts
ema-pytorch
torch>=1.6
torchvision
transformers
vector-quantize-pytorch>=0.7.3
