chore: move clip models

server/clip_server/model/pretrained_models.py

@@ -154,8 +154,8 @@
             response = requests.head(hf_download_url, timeout=10)
             if response.status_code in [200, 302]:
                 return (hf_download_url, model_pretrained[1])
         except Exception:
             raise ValueError('Invalid model url.')
 
 
 def download_model(
@@ -168,6 +168,8 @@
     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)
 

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