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

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

[eval]
scipy
torchvision
nltk
inflection
pycocoevalcap
pycocotools
tqdm

[training]
wandb
torchvision
braceexpand
webdataset
tqdm
