reimagine-it / requirements.txt
Alberto Carmona
Add the meshed-memory-transformer
118c782
# 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