ltttpku / CMMP

14 stars 0 forks source link

About pretrained CLIP weight #9

Closed YangJae96 closed 1 week ago

YangJae96 commented 1 week ago

Hi. Thank you for your great work.

How can I download the pretrained CLIP, ViT-B-16.pt?

In OpenAI, CLIP github, they only bring the VIT-B-16 version with the codes.

import clip
import torch

model, preprocess = clip.load("ViT-B/16", device="cuda" if torch.cuda.is_available() else "cpu")

I cant figure out how to download ViT-B/16.pt weights and save it in the local.

Can I get some help?

ltttpku commented 1 week ago

Thanks for your interest in our work. The CLIP weights can be downloaded from the URLs here.

YangJae96 commented 1 week ago

@ltttpku Thanks!