Extend sentence-transformers ONNX test (#1734)

* extend sentence transformers onnx test * trigger ci
huggingface · Mar 18, 2024 · 23fe6f7 · 23fe6f7
1 parent 9ff5ea8
commit 23fe6f7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/exporters/exporters_utils.py b/tests/exporters/exporters_utils.py
@@ -334,7 +334,10 @@
 
 PYTORCH_SENTENCE_TRANSFORMERS_MODEL = {
     "clip": "sentence-transformers/clip-ViT-B-32",
-    "transformer": "sentence-transformers/all-MiniLM-L6-v2",
+    "transformer": {
+        "sentence-transformers/all-MiniLM-L6-v2": ["feature-extraction", "sentence-similarity"],
+        "fxmarty/tiny-dummy-mistral-sentence-transformer": ["feature-extraction", "sentence-similarity"],
+    },
 }