# For image generation
git+https://github.com/patil-suraj/transformers@ldm-bert
torch
diffusers==0.1.1
opencv-python

# For captioning
git+https://github.com/ruotianluo/meshed-memory-transformer
transformers
h5py
pandas
tqdm
scikit-image
ipywidgets
wandb
bert-score
ftfy
timm
lmdbdict
yacs
pyemd
gensim
pytorch-lightning==1.0.0
gdown