accelerate
einops>=0.4
local-attention
torch>=1.6
vector-quantize-pytorch>=1.1.5
