accelerate==0.10.0
click
einops>=0.4
einops-exts
ema-pytorch>=0.0.3
numpy
packaging
pillow
pydantic
pytorch-lightning
pytorch-warmup
resize-right
sentencepiece
torch>=1.6
torchvision
transformers
tqdm
