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