Update README.md
Browse files
README.md
CHANGED
@@ -12,7 +12,7 @@ from PIL import Image
|
|
12 |
import open_clip
|
13 |
|
14 |
##load the model
|
15 |
-
model, _, preprocess = open_clip.create_model_and_transforms('ViT-B-16', pretrained='/
|
16 |
cache_dir='/mnt/Xsky/syx/model/open_clip', force_quick_gelu=True)
|
17 |
tokenizer = open_clip.get_tokenizer('ViT-B-16')
|
18 |
model = model.cuda()
|
|
|
12 |
import open_clip
|
13 |
|
14 |
##load the model
|
15 |
+
model, _, preprocess = open_clip.create_model_and_transforms('ViT-B-16', pretrained='your_path/pathclip-base.pt',
|
16 |
cache_dir='/mnt/Xsky/syx/model/open_clip', force_quick_gelu=True)
|
17 |
tokenizer = open_clip.get_tokenizer('ViT-B-16')
|
18 |
model = model.cuda()
|