Skip to content

Commit 1c50a5f

Browse files
authored
[tests] enforce torch version in the compilation tests. (#11979)
enforce torch version in the compilation tests.
1 parent 7ae6347 commit 1c50a5f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/models/test_modeling_common.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1948,6 +1948,7 @@ def test_push_to_hub_library_name(self):
19481948
@require_torch_2
19491949
@is_torch_compile
19501950
@slow
1951+
@require_torch_version_greater("2.7.1")
19511952
class TorchCompileTesterMixin:
19521953
different_shapes_for_compilation = None
19531954

@@ -2046,6 +2047,7 @@ def test_compile_on_different_shapes(self):
20462047
@require_torch_accelerator
20472048
@require_peft_backend
20482049
@require_peft_version_greater("0.14.0")
2050+
@require_torch_version_greater("2.7.1")
20492051
@is_torch_compile
20502052
class LoraHotSwappingForModelTesterMixin:
20512053
"""Test that hotswapping does not result in recompilation on the model directly.

0 commit comments

Comments
 (0)