einops>=0.7
einops-exts
ema-pytorch
torch>=1.6
torchvision
transformers
vector-quantize-pytorch>=1.11.8
