Merge pull request #137 from deepghs/dev/siglip

dev(narugo): more test cases for siglip
deepghs · Jan 29, 2025 · 44e38ed · 44e38ed
2 parents 5a407c5 + c74017e
commit 44e38ed
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/test/preprocess/transformers/test_siglip.py b/test/preprocess/transformers/test_siglip.py
@@ -23,11 +23,8 @@ class TestPreprocessTransformersSiglip:
         'repo_id': [
             'Marqo/marqo-ecommerce-embeddings-B',
             'ucsahin/TraVisionLM-DPO',
-            'google/siglip-base-patch16-384',
-            'google/siglip-base-patch16-512',
             'llava-hf/llava-interleave-qwen-0.5b-hf',
             'zhumj34/Mipha-3B',
-            'google/siglip-so400m-patch14-384',
             'lmms-lab/llava-onevision-qwen2-72b-ov-sft',
             'p1atdev/siglip-tagger-test-3',
             'gokaygokay/paligemma-rich-captions',
@@ -36,6 +33,17 @@ class TestPreprocessTransformersSiglip:
             'OpenFace-CQUPT/Human_LLaVA',
             'ucsahin/TraVisionLM-base',
             'mlx-community/paligemma-3b-mix-448-8bit',
+
+            'google/siglip-so400m-patch14-384',
+            'google/siglip-so400m-patch14-224',
+            'google/siglip-base-patch16-256-multilingual',
+            'google/siglip-base-patch16-224',
+            'google/siglip-so400m-patch16-256-i18n',
+            'google/siglip-base-patch16-512',
+            'google/siglip-large-patch16-256',
+            'google/siglip-base-patch16-384',
+            'google/siglip-large-patch16-384',
+            'google/siglip-base-patch16-256'
         ],
         'src_image': [
             'png_640.png',