add test for sentence transformers

huggingface · Feb 13, 2024 · 6d1dc0a · 6d1dc0a
1 parent b2751dc
commit 6d1dc0a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
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",