einops>=0.5
ema-pytorch
torch>=1.6
transformers
vector-quantize-pytorch>=0.10.5
