M-CLIP
/

XLM-Roberta-Large-Vit-B-16Plus

Lauler commited on Jun 3, 2022

Commit

ed54b77

•

1 Parent(s): e092132

Specify pretrained model version

Files changed (1) hide show

README.md CHANGED Viewed

@@ -48,7 +48,7 @@ import requests
 from PIL import Image
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model, _, preprocess = open_clip.create_model_and_transforms('ViT-B-16-plus-240')
 model.to(device)
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"

 from PIL import Image
 device = "cuda" if torch.cuda.is_available() else "cpu"
+model, _, preprocess = open_clip.create_model_and_transforms('ViT-B-16-plus-240', pretrained="laion400m_e32")
 model.to(device)
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"