fix: download bugs

jina-ai · Jan 22, 2024 · bc722cd · bc722cd
1 parent ff49880
commit bc722cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/server/clip_server/model/pretrained_models.py b/server/clip_server/model/pretrained_models.py
@@ -168,6 +168,8 @@ def download_model(
     os.makedirs(target_folder, exist_ok=True)
     filename = os.path.basename(url)
     filename = filename.split('?')[0]
+    if filename.split('.')[-1] == 'onnx':
+        filename = filename.split('-')[-1]
 
     download_target = os.path.join(target_folder, filename)
 

diff --git a/server/clip_server/torch-flow.yml b/server/clip_server/torch-flow.yml
@@ -8,6 +8,6 @@ executors:
       jtype: CLIPEncoder
       metas:
         py_modules:
-          - clip_server.executors.clip_torch
+          - clip_server.executors.clip_onnx
     timeout_ready: 3000000
     replicas: 1