Where can I download the file 'ViT-L-14_state_dict.pt'?
#1
by
YUkI51121
- opened
I have no idea about the source of this file...... :(
may I should use the code below to load the pretrained model?
model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms('ViT-L-14', pretrained='openai', cache_dir='the path of pretrained model')
Hello, have you found this parameter
import open_clip
import torch
model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms('ViT-L-14')
weights_path = '/old_home/lyt/zxj_workplaces/PMC-CLIP/open_clip_pytorch_model.bin'
model.load_state_dict(torch.load(weights_path))
tokenizer = open_clip.get_tokenizer('ViT-L-14')
print(model)
torch.save(model.state_dict(), 'pmc_vit_l_14_weights.pt')
you can download it to local and run this program to make it a pt files.