Skip to content

Commit 4d82111

Browse files
Add version guards to tests
1 parent 716f446 commit 4d82111

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

tests/models/test_modeling_common.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@
6363
is_torch_compile,
6464
numpy_cosine_similarity_distance,
6565
require_peft_backend,
66+
require_peft_version_greater,
6667
require_torch_2,
6768
require_torch_accelerator,
6869
require_torch_accelerator_with_training,
@@ -1670,6 +1671,7 @@ def test_push_to_hub_library_name(self):
16701671
@require_torch_2
16711672
@require_torch_accelerator
16721673
@require_peft_backend
1674+
@require_peft_version_greater("0.14.0")
16731675
@is_torch_compile
16741676
class TestLoraHotSwappingForModel(unittest.TestCase):
16751677
"""Test that hotswapping does not result in recompilation on the model directly.

tests/pipelines/test_pipelines.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,7 @@
8181
require_hf_hub_version_greater,
8282
require_onnxruntime,
8383
require_peft_backend,
84+
require_peft_version_greater,
8485
require_torch_2,
8586
require_torch_accelerator,
8687
require_transformers_version_greater,
@@ -2184,6 +2185,7 @@ def test_ddpm_ddim_equality_batched(self):
21842185
@require_torch_2
21852186
@require_torch_accelerator
21862187
@require_peft_backend
2188+
@require_peft_version_greater("0.14.0")
21872189
@is_torch_compile
21882190
class TestLoraHotSwappingForPipeline(unittest.TestCase):
21892191
"""Test that hotswapping does not result in recompilation in a pipeline.

0 commit comments

Comments
 (0)