accelerate
beartype
click
datasets
einops>=0.6
einops-exts
ema-pytorch>=0.0.3
fsspec
kornia
lion-pytorch
numpy
packaging
pillow
pydantic
pytorch-lightning
pytorch-warmup
sentencepiece
torch>=1.6
torchvision
transformers
triton
tqdm
