localhorst/CLIP_on_Tesla_K20Xm
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
e1b7bac3ec
CLIP_on_Tesla_K20Xm/tests/test_encode.py
Jacky e1b7bac3ec Add .gitignore and installable module
2021-01-26 17:30:35 +11:00

5 lines
97 B
Python
Raw Blame History

import clip
import torch
device = 'cpu'
model, preprocess = clip.load("ViT-B/32", device=device)
Reference in New Issue View Git Blame Copy Permalink