CLIP-Interrogator / requirements.txt
osanseviero's picture
CUDA error fix attempt (#3)
57c304a
raw
history blame
245 Bytes
--extra-index-url https://download.pytorch.org/whl/cu113
torch
torchvision
fairscale
ftfy
Pillow
timm
transformers==4.15.0
-e git+https://github.com/openai/CLIP.git@main#egg=clip
-e git+https://github.com/pharmapsychotic/BLIP.git@main#egg=blip