Skip to content

Commit 4cf5a56

Browse files
committed
fix
1 parent a0ab39c commit 4cf5a56

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

_unittests/ut_torch_models/test_validate_whole_models1.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,7 @@ def test_g_validate_model_onnx_dynamo_os_ort(self):
122122
@hide_stdout()
123123
@ignore_warnings(FutureWarning)
124124
@requires_experimental()
125+
@requires_transformers("4.52") # fix it
125126
def test_i_validate_model_custom(self):
126127
mid = "arnir0/Tiny-LLM"
127128
summary, data = validate_model(
@@ -150,6 +151,7 @@ def test_i_validate_model_custom(self):
150151
@requires_torch("2.7")
151152
@hide_stdout()
152153
@ignore_warnings(FutureWarning)
154+
@requires_transformers("4.52")
153155
@requires_experimental()
154156
def test_j_validate_model_custom_torch(self):
155157
mid = "arnir0/Tiny-LLM"

0 commit comments

Comments
 (0)