file

Author: xadupre

_unittests/ut_torch_models/test_validate_whole_models.py

@@ -270,7 +270,6 @@ def test_validate_phi35_mini_instruct(self):
             inputs2=True,
             patch=True,
             rewrite=True,
-            # model_options={"rope_scaling": {"rope_type": "dynamic", "factor": 10.0}},
         )
         self.assertIsInstance(summary, dict)
         self.assertIsInstance(data, dict)
@@ -279,6 +278,29 @@ def test_validate_phi35_mini_instruct(self):
         op_types = set(n.op_type for n in onx.graph.node)
         self.assertIn("If", op_types)
 
+    @requires_torch("2.7")
+    @hide_stdout()
+    @ignore_warnings(FutureWarning)
+    @requires_transformers("4.51")
+    def test_validate_phi35_4k_mini_instruct(self):
+        mid = "microsoft/Phi-3.5-mini-4k-instruct"
+        summary, data = validate_model(
+            mid,
+            do_run=True,
+            verbose=10,
+            exporter="custom",
+            dump_folder="dump_test/validate_phi35_mini_instruct",
+            inputs2=True,
+            patch=True,
+            rewrite=True,
+            model_options={"rope_scaling": {"rope_type": "dynamic", "factor": 10.0}},
+        )
+        self.assertIsInstance(summary, dict)
+        self.assertIsInstance(data, dict)
+        onnx_filename = data["onnx_filename"]
+        onx = onnx.load(onnx_filename)
+        op_types = set(n.op_type for n in onx.graph.node)
+        self.assertIn("If", op_types)
 
 if __name__ == "__main__":
     unittest.main(verbosity=2)