einops
einops-exts
transformers
peft
bigmodelvis
torch
torchvision
pillow
more-itertools
datasets
braceexpand
webdataset
wandb
nltk
scipy
inflection
sentencepiece
open_clip_torch
mmengine
gradio
