diff --git a/_unittests/ut_torch_models/test_validate_models.py b/_unittests/ut_torch_models/test_validate_models.py index d6b0b07d..dcb8112f 100644 --- a/_unittests/ut_torch_models/test_validate_models.py +++ b/_unittests/ut_torch_models/test_validate_models.py @@ -33,6 +33,27 @@ def test_validate_microsoft_phi4_reasoning(self): self.assertLess(summary["disc_onnx_ort_run_abs"], 1e-5) self.assertIn("onnx_filename", data) + @requires_transformers("4.52") + @requires_torch("2.7.99") + @requires_experimental() + @hide_stdout() + def test_validate_microsoft_phi3_mini_128k(self): + # python -m onnx_diagnostic validate -m microsoft/Phi-3-mini-128k-instruct + # --run -v 1 --export custom -o dump_test --no-quiet --device cuda --patch + summary, data = validate_model( + "microsoft/Phi-3-mini-128k-instruct", + do_run=True, + verbose=2, + exporter="custom", + do_same=True, + patch=True, + rewrite=True, + stop_if_static=2 if pv.Version(torch.__version__) > pv.Version("2.6.1") else 0, + dump_folder="dump_test/validate_microsoft_phi3_mini_128k", + ) + self.assertLess(summary["disc_onnx_ort_run_abs"], 1e-5) + self.assertIn("onnx_filename", data) + if __name__ == "__main__": unittest.main(verbosity=2)