fix

Author: xadupre

_unittests/ut_torch_models/test_test_helpers.py

@@ -1,7 +1,7 @@
 import copy
 import unittest
 import packaging.version as pv
-import transformers
+import torch
 from onnx_diagnostic.ext_test_case import ExtTestCase, hide_stdout, ignore_warnings
 from onnx_diagnostic.torch_models.test_helper import (
     get_inputs_for_task,
@@ -64,7 +64,7 @@ def test_validate_model_onnx(self):
             exporter="onnx-dynamo",
             dump_folder="dump_test_validate_model_onnx",
             patch=True,
-            stop_if_static=pv.Version(transformers.__version__) >= pv.Version("4.50"),
+            stop_if_static=pv.Version(torch.__version__) > pv.Version("2.6"),
         )
         self.assertIsInstance(summary, dict)
         self.assertIsInstance(data, dict)