diff --git a/tests/models/test_modeling_common.py b/tests/models/test_modeling_common.py index 8309700ce106..435bd32c6083 100644 --- a/tests/models/test_modeling_common.py +++ b/tests/models/test_modeling_common.py @@ -1948,6 +1948,7 @@ def test_push_to_hub_library_name(self): @require_torch_2 @is_torch_compile @slow +@require_torch_version_greater("2.7.1") class TorchCompileTesterMixin: different_shapes_for_compilation = None @@ -2046,6 +2047,7 @@ def test_compile_on_different_shapes(self): @require_torch_accelerator @require_peft_backend @require_peft_version_greater("0.14.0") +@require_torch_version_greater("2.7.1") @is_torch_compile class LoraHotSwappingForModelTesterMixin: """Test that hotswapping does not result in recompilation on the model directly.