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

[all]
torch==2.0.1
nltk
transformers>=4.28.1
open_clip_torch>=2.16.0
sentencepiece==0.1.98
scipy
pycocoevalcap
einops-exts
wandb
braceexpand
webdataset
torchvision
tqdm
pycocotools
inflection
einops
pillow

[eval]
scipy
torchvision
nltk
inflection
pycocoevalcap
pycocotools
tqdm

[training]
wandb
torchvision
braceexpand
webdataset
tqdm
