add test for sentence transformers

echarlaix · echarlaix · commit 6d1dc0aa6960 · 2024-02-13T15:55:44.000+01:00
diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py
@@ -426,6 +426,7 @@ class OVModelForFeatureExtractionIntegrationTest(unittest.TestCase):
         "bert",
         "distilbert",
         "roberta",
+        "sentence-transformers-bert",
     )
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES)
diff --git a/tests/openvino/utils_tests.py b/tests/openvino/utils_tests.py
@@ -70,6 +70,7 @@
     "roberta": "hf-internal-testing/tiny-random-roberta",
     "roformer": "hf-internal-testing/tiny-random-roformer",
     "segformer": "hf-internal-testing/tiny-random-SegformerModel",
+    "sentence-transformers-bert": "sentence-transformers-testing/stsb-bert-tiny-safetensors"
     "speech_to_text": "hf-internal-testing/tiny-random-Speech2TextModel",
     "squeezebert": "hf-internal-testing/tiny-random-squeezebert",
     "stable-diffusion": "hf-internal-testing/tiny-stable-diffusion-torch",

Original file line number	Diff line number	Diff line change
`@@ -426,6 +426,7 @@ class OVModelForFeatureExtractionIntegrationTest(unittest.TestCase):`
`426`	`426`	`"bert",`
`427`	`427`	`"distilbert",`
`428`	`428`	`"roberta",`
	`429`	`+ "sentence-transformers-bert",`
`429`	`430`	`)`
`430`	`431`
`431`	`432`	`@parameterized.expand(SUPPORTED_ARCHITECTURES)`