fix: update trt model list

jina-ai · Aug 3, 2022 · 2c8629d · 2c8629d
1 parent d4f6155
commit 2c8629d
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/server/clip_server/model/clip_trt.py b/server/clip_server/model/clip_trt.py
@@ -21,6 +21,20 @@
 from clip_server.model.clip_onnx import _MODELS as ONNX_MODELS
 
 _MODELS = [
+    'RN50::openai',
+    'RN50::yfcc15m',
+    'RN50::cc12m',
+    'RN101::openai',
+    'RN101::yfcc15m',
+    'RN50x4::openai',
+    'ViT-B-32::openai',
+    'ViT-B-32::laion2b_e16',
+    'ViT-B-32::laion400m_e31',
+    'ViT-B-32::laion400m_e32',
+    'ViT-B-16::openai',
+    'ViT-B-16::laion400m_e31',
+    'ViT-B-16::laion400m_e32',
+    # older version name format
     'RN50',
     'RN101',
     'RN50x4',