accelerate
beartype
einops>=0.6
ema-pytorch
pillow
sentencepiece
torch>=1.6
transformers
torch>=1.6
torchvision
tqdm
vector-quantize-pytorch>=0.10.14
