update clip

Signed-off-by: Filip Haltmayer <filip.haltmayer@zilliz.com>
towhee-io · Jan 25, 2023 · 85dfed1 · 85dfed1
1 parent 447e1b3
commit 85dfed1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/image_text_embedding/description.yaml b/image_text_embedding/description.yaml
@@ -48,10 +48,10 @@ models:
         op-name: "image_text_embedding.clip"
         op-link: "https://towhee.io/image-text-embedding/clip"
         pretrained-models:
-            - "image_text_embedding.clip(model_name='clip_resnet_r50')"
-            - "image_text_embedding.clip(model_name='clip_resnet_r101')"
-            - "image_text_embedding.clip(model_name='clip_vit_b16')"
-            - "image_text_embedding.clip(model_name='clip_vit_b32')"
+            - "image_text_embedding.clip(model_name='clip_vit_base_patch16')"
+            - "image_text_embedding.clip(model_name='clip_vit_base_patch32')"
+            - "image_text_embedding.clip(model_name='clip_vit_large_patch14')"
+            - "image_text_embedding.clip(model_name='clip_vit_large_patch14_336')"
     - blip:
         model: "BLIP"
         op-name: "image_text_embedding.blip"