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