localhorst/CLIP_on_Tesla_K20Xm
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add .gitignore and installable module

Browse Source
This commit is contained in:
Jacky
2021-01-26 17:30:35 +11:00
parent d5ba15cf46
commit e1b7bac3ec
4 changed files with 145 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_encode.py Normal file
View File

@ -0,0 +1,4 @@
import clip
import torch
device = 'cpu'
model, preprocess = clip.load("ViT-B/32", device=device)