Unverified Commit 1c50a5f7 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[tests] enforce torch version in the compilation tests. (#11979)

enforce torch version in the compilation tests.
parent 7ae6347e
...@@ -1948,6 +1948,7 @@ class ModelPushToHubTester(unittest.TestCase): ...@@ -1948,6 +1948,7 @@ class ModelPushToHubTester(unittest.TestCase):
@require_torch_2 @require_torch_2
@is_torch_compile @is_torch_compile
@slow @slow
@require_torch_version_greater("2.7.1")
class TorchCompileTesterMixin: class TorchCompileTesterMixin:
different_shapes_for_compilation = None different_shapes_for_compilation = None
...@@ -2046,6 +2047,7 @@ class TorchCompileTesterMixin: ...@@ -2046,6 +2047,7 @@ class TorchCompileTesterMixin:
@require_torch_accelerator @require_torch_accelerator
@require_peft_backend @require_peft_backend
@require_peft_version_greater("0.14.0") @require_peft_version_greater("0.14.0")
@require_torch_version_greater("2.7.1")
@is_torch_compile @is_torch_compile
class LoraHotSwappingForModelTesterMixin: class LoraHotSwappingForModelTesterMixin:
"""Test that hotswapping does not result in recompilation on the model directly. """Test that hotswapping does not result in recompilation on the model directly.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment