Skip to content

Commit fb27894

Browse files
committed
fix tests
1 parent 34db69b commit fb27894

File tree

1 file changed

+12
-7
lines changed

1 file changed

+12
-7
lines changed

tests/openvino/test_modeling.py

+12-7
Original file line numberDiff line numberDiff line change
@@ -980,7 +980,6 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
980980
if is_transformers_version(">=", "4.40.0"):
981981
SUPPORTED_ARCHITECTURES += (
982982
"gemma",
983-
"gemma2",
984983
"olmo",
985984
"stablelm",
986985
"starcoder2",
@@ -990,14 +989,20 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
990989
"qwen2",
991990
"qwen2-moe",
992991
"arctic",
993-
"exaone",
994-
"mistral-nemo",
995-
"minicpm3",
996-
"glm",
997-
"granite",
998-
"granite-moe",
999992
)
1000993

994+
if is_transformers_version(">=", "4.41.0"):
995+
SUPPORTED_ARCHITECTURES += ("phi3",)
996+
997+
if is_transformers_version(">=", "4.43.0"):
998+
SUPPORTED_ARCHITECTURES += ("gemma2", "exaone")
999+
1000+
if is_transformers_version(">=", "4.44.0"):
1001+
SUPPORTED_ARCHITECTURES += ("granite", "granite-moe")
1002+
1003+
if is_transformers_version(">=", "4.46.0"):
1004+
SUPPORTED_ARCHITECTURES += ("glm", "mistral-nemo", "minicpm3")
1005+
10011006
# gptq and awq install disabled for windows test environment
10021007
if platform.system() != "Windows":
10031008
SUPPORTED_ARCHITECTURES += ("opt_gptq",)

0 commit comments

Comments
 (0)