Skip to content

Commit 393d391

Browse files
committed
fix CIs - 4.48.3
1 parent 9568e18 commit 393d391

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

_unittests/ut_tasks/test_tasks.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ def test_text2text_generation(self):
3535
)
3636

3737
@hide_stdout()
38+
@requires_transformers("4.55.4") # modeling_units
3839
def test_text_generation(self):
3940
mid = "arnir0/Tiny-LLM"
4041
data = get_untrained_model_with_inputs(mid, verbose=1, add_second_input=True)

_unittests/ut_torch_models/test_validate_whole_models.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,7 @@ def test_g_validate_model_onnx_dynamo_os_ort(self):
118118
self.assertExists(onnx_filename)
119119

120120
@requires_torch("2.7")
121+
@requires_transformers("4.55.4") # modeling_units
121122
@hide_stdout()
122123
@ignore_warnings(FutureWarning)
123124
@requires_experimental()
@@ -147,6 +148,7 @@ def test_i_validate_model_custom(self):
147148
)
148149

149150
@requires_torch("2.7")
151+
@requires_transformers("4.55.4") # modeling_units
150152
@hide_stdout()
151153
@ignore_warnings(FutureWarning)
152154
@requires_experimental()

0 commit comments

Comments
 (0)