Skip to content

Commit 8aa3802

Browse files
committed
change switch version
1 parent 2622235 commit 8aa3802

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

_unittests/ut_torch_models/test_tiny_llms_onnx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def test_onnx_export_tiny_llm_xdbg(self):
7070

7171
@ignore_warnings((UserWarning, DeprecationWarning, FutureWarning))
7272
@hide_stdout()
73-
@requires_torch("2.10") # this test broke on CI but works locally
73+
@requires_torch("2.10.99") # this test broke on CI but works locally
7474
def test_bypass_onnx_export_tiny_llm_official_nopositionids(self):
7575
data = get_tiny_llm()
7676
model, inputs, ds = data["model"], data["inputs"], data["dynamic_shapes"]

_unittests/ut_torch_models/test_validate_whole_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def test_e_validate_model_export(self):
7171
self.assertIsInstance(data, dict)
7272

7373
@requires_torch("2.8.99")
74-
@requires_transformers("4.51")
74+
@requires_transformers("4.54")
7575
@hide_stdout()
7676
@ignore_warnings(FutureWarning)
7777
def test_f_validate_model_onnx_dynamo_ir(self):

0 commit comments

Comments
 (0)