fix: add :: to model name

jina-ai · Nov 16, 2022 · c4beeca · c4beeca
1 parent 7fcb813
commit c4beeca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/server/clip_server/model/openclip_model.py b/server/clip_server/model/openclip_model.py
@@ -27,7 +27,6 @@ def __init__(self, name: str, device: str = 'cpu', jit: bool = False, **kwargs):
         # model_url, md5sum = get_model_url_md5(name)
         # model_path = download_model(model_url, md5sum=md5sum)
         model_path = '/home/zonlin/.cache/clip/test.bin'
-        pretrained = None
 
         if pretrained == 'openai':
             self._model = load_openai_model(model_path, device=device, jit=jit)

diff --git a/server/clip_server/model/pretrained_models.py b/server/clip_server/model/pretrained_models.py
@@ -70,11 +70,11 @@
         'ViT-g-14-laion2b-s12b-b42k.bin',
         '3bf99353f6f1829faac0bb155be4382a',
     ),
-    'roberta-ViT-B-32': (
+    'roberta-ViT-B-32::laion2b-s12b-b32k': (
         'ViT-B-32-laion2b-s12b-b32k.bin',
         '76d4c9d13774cc15fa0e2b1b94a8402c',
     ),
-    'xlm-roberta-base-ViT-B-32': (
+    'xlm-roberta-base-ViT-B-32::laion5b-s13b-b90k': (
         'ViT-B-32::laion5b-s13b-b90k.bin',
         'f68abc07ef349720f1f880180803142d',
     ),