Skip to content

Commit 7e8fb8e

Browse files
committed
add test
1 parent 2436a9f commit 7e8fb8e

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

tests/openvino/test_modeling.py

+16
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,23 @@ def __exit__(self, type, value, traceback):
105105

106106
class OVModelForMaskedLMIntegrationTest(unittest.TestCase):
107107
SUPPORTED_ARCHITECTURES = (
108+
"albert",
108109
"bert",
110+
# "camembert",
111+
# "convbert",
112+
# "data2vec_text",
113+
"deberta",
114+
# "deberta_v2",
115+
"distilbert",
116+
"electra",
117+
"flaubert",
118+
"ibert",
119+
# "mobilebert",
120+
"roberta",
121+
"roformer",
122+
"squeezebert",
123+
"xlm",
124+
"xlm_roberta",
109125
)
110126

111127
@parameterized.expand(SUPPORTED_ARCHITECTURES)

0 commit comments

Comments
 (0)