Extend sentence-transformers ONNX test (#1734)

fxmarty · web-flow · commit 23fe6f75974e · 2024-03-18T16:05:03.000+08:00
* extend sentence transformers onnx test

* trigger ci
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"],
+    },
 }
 
 

Original file line number	Diff line number	Diff line change
`@@ -334,7 +334,10 @@`
`334`	`334`
`335`	`335`	`PYTORCH_SENTENCE_TRANSFORMERS_MODEL = {`
`336`	`336`	`"clip": "sentence-transformers/clip-ViT-B-32",`
`337`		`- "transformer": "sentence-transformers/all-MiniLM-L6-v2",`
	`337`	`+ "transformer": {`
	`338`	`+ "sentence-transformers/all-MiniLM-L6-v2": ["feature-extraction", "sentence-similarity"],`
	`339`	`+ "fxmarty/tiny-dummy-mistral-sentence-transformer": ["feature-extraction", "sentence-similarity"],`
	`340`	`+ },`
`338`	`341`	`}`
`339`	`342`
`340`	`343`