Commit
·
c41ff17
1
Parent(s):
fb16ac2
Update README.md
Browse files
README.md
CHANGED
@@ -30,7 +30,7 @@ print("Available models:", available_models())
|
|
30 |
# Available models: ['ViT-B-16', 'ViT-L-14', 'ViT-L-14-336', 'ViT-H-14', 'RN50']
|
31 |
|
32 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
33 |
-
model, preprocess = load_from_name("ViT-
|
34 |
model.eval()
|
35 |
image = preprocess(Image.open("examples/pokemon.jpeg")).unsqueeze(0).to(device)
|
36 |
text = clip.tokenize(["杰尼龟", "妙蛙种子", "小火龙", "皮卡丘"]).to(device)
|
|
|
30 |
# Available models: ['ViT-B-16', 'ViT-L-14', 'ViT-L-14-336', 'ViT-H-14', 'RN50']
|
31 |
|
32 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
33 |
+
model, preprocess = load_from_name("ViT-L-14", device=device, download_root='./')
|
34 |
model.eval()
|
35 |
image = preprocess(Image.open("examples/pokemon.jpeg")).unsqueeze(0).to(device)
|
36 |
text = clip.tokenize(["杰尼龟", "妙蛙种子", "小火龙", "皮卡丘"]).to(device)
|