add test

huggingface · Feb 22, 2024 · 7e8fb8e · 7e8fb8e
1 parent 2436a9f
commit 7e8fb8e
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py
@@ -105,7 +105,23 @@ def __exit__(self, type, value, traceback):
 
 class OVModelForMaskedLMIntegrationTest(unittest.TestCase):
     SUPPORTED_ARCHITECTURES = (
+        "albert",
         "bert",
+        # "camembert",
+        # "convbert",
+        # "data2vec_text",
+        "deberta",
+        # "deberta_v2",
+        "distilbert",
+        "electra",
+        "flaubert",
+        "ibert",
+        # "mobilebert",
+        "roberta",
+        "roformer",
+        "squeezebert",
+        "xlm",
+        "xlm_roberta",
     )
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES)