einops>=0.4
ema-pytorch
torch>=1.6
vector-quantize-pytorch>=0.10.1
