Add onnxruntime inference test for phi model

echarlaix · echarlaix · commit 68f2c4ef6951 · 2024-03-21T16:36:12.000+01:00
diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
@@ -2258,6 +2258,7 @@ class ORTModelForCausalLMIntegrationTest(ORTModelTestMixin):
         "llama",
         "mistral",
         "mpt",
+        "phi",
         "qwen2",
     ]
 
diff --git a/tests/onnxruntime/utils_onnxruntime_tests.py b/tests/onnxruntime/utils_onnxruntime_tests.py
@@ -128,6 +128,7 @@
     "pegasus": "hf-internal-testing/tiny-random-PegasusModel",
     "perceiver_text": "hf-internal-testing/tiny-random-language_perceiver",
     "perceiver_vision": "hf-internal-testing/tiny-random-vision_perceiver_conv",
+    "phi": "hf-internal-testing/tiny-random-PhiForCausalLM",
     "pix2struct": "fxmarty/pix2struct-tiny-random",
     "poolformer": "hf-internal-testing/tiny-random-PoolFormerModel",
     "qwen2": "fxmarty/tiny-dummy-qwen2",

Original file line number	Diff line number	Diff line change
`@@ -2258,6 +2258,7 @@ class ORTModelForCausalLMIntegrationTest(ORTModelTestMixin):`
`2258`	`2258`	`"llama",`
`2259`	`2259`	`"mistral",`
`2260`	`2260`	`"mpt",`
	`2261`	`+ "phi",`
`2261`	`2262`	`"qwen2",`
`2262`	`2263`	`]`
`2263`	`2264`