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()
|