Add phi OpenVINO export support

echarlaix · echarlaix · commit 56b152e7b107 · 2024-03-21T15:53:23.000+01:00
diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py
@@ -502,6 +502,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
         "qwen2",
         "stablelm",
         # "starcoder2", # TODO: enable with next transformers release
+        "phi",
     )
     GENERATION_LENGTH = 100
     IS_SUPPORT_STATEFUL = is_openvino_version(">=", "2023.3")
diff --git a/tests/openvino/utils_tests.py b/tests/openvino/utils_tests.py
@@ -70,6 +70,7 @@
     "nystromformer": "hf-internal-testing/tiny-random-NystromformerModel",
     "pegasus": "hf-internal-testing/tiny-random-pegasus",
     "pix2struct": "fxmarty/pix2struct-tiny-random",
+    "phi": "hf-internal-testing/tiny-random-PhiForCausalLM",
     "poolformer": "hf-internal-testing/tiny-random-PoolFormerModel",
     "qwen": "katuni4ka/tiny-random-qwen",
     "qwen2": "Qwen/Qwen1.5-0.5B",

Original file line number	Diff line number	Diff line change
`@@ -502,6 +502,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):`
`502`	`502`	`"qwen2",`
`503`	`503`	`"stablelm",`
`504`	`504`	`# "starcoder2", # TODO: enable with next transformers release`
	`505`	`+ "phi",`
`505`	`506`	`)`
`506`	`507`	`GENERATION_LENGTH = 100`
`507`	`508`	`IS_SUPPORT_STATEFUL = is_openvino_version(">=", "2023.3")`