accelerate
beartype>=0.16.1
einops>=0.7.0
ema-pytorch>=0.2.2
encodec
fairseq
joblib
lion-pytorch
local-attention>=1.8.4
scikit-learn
sentencepiece
torch>=1.12
torchaudio
transformers
tqdm
vector-quantize-pytorch>=1.7.0
