requirements.txt

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

# # For captioning
# transformers
# h5py
# pandas
# tqdm
# scikit-image
# ipywidgets
# wandb
# bert-score
# ftfy
# timm
# lmdbdict
# yacs
# pyemd
# gensim
# pytorch-lightning==1.0.0
# gdown