DavidHuji / CapDec

CapDec: SOTA Zero Shot Image Captioning Using CLIP and GPT2, EMNLP 2022 (findings)
MIT License
181 stars 19 forks source link

Requirements File #2

Open Adesoji1 opened 1 year ago

Adesoji1 commented 1 year ago

This Requirements file was generated on a windows 11 machine, virtual environment python 3.8. Kindly view and add to the requirements.txt file requirements.txt

dinhanhx commented 1 year ago

You should not include as an attachment. Just list in the comment

clip==0.2.0
clip_anytorch==2.5.0
matplotlib==3.3.3
numpy==1.23.4
Pillow==9.3.0
pycocotools==2.0.5
scikit_image==0.19.3
skimage==0.0
torch==1.13.0
torchvision==0.14.0
tqdm==4.64.1
transformers==4.24.0
wandb==0.13.5
Adesoji1 commented 1 year ago

This is the requirements.txt content below. Kindly consider and vet in your main branch

clip==0.2.0 clip_anytorch==2.5.0 matplotlib==3.3.3 numpy==1.23.4 Pillow==9.3.0 pycocotools==2.0.5 scikit_image==0.19.3 skimage==0.0 torch==1.13.0 torchvision==0.14.0 tqdm==4.64.1 transformers==4.24.0 wandb==0.13.5

DavidHuji commented 1 year ago

Thanks! I'll add that.