requires.txt

einops
einops-exts
transformers
torch>=2.0.1
pillow
open_clip_torch>=2.16.0
sentencepiece

[all]
sentencepiece
braceexpand
transformers
torch>=2.0.1
tqdm
einops
webdataset
torchvision
pillow
einops-exts
open_clip_torch>=2.16.0
wandb

[training]
torchvision
braceexpand
webdataset
tqdm
wandb