Skip to content

Commit fb5f5f7

Browse files
committed
Add back skip_mps.
1 parent a20d03d commit fb5f5f7

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/lora/utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@
3939
require_peft_version_greater,
4040
require_transformers_version_greater,
4141
torch_device,
42+
skip_mps,
4243
)
4344

4445

@@ -1562,6 +1563,7 @@ def test_simple_inference_with_text_denoiser_multi_adapter_weighted(self):
15621563
"output with no lora and output with lora disabled should give same results",
15631564
)
15641565

1566+
@skip_mps
15651567
@pytest.mark.xfail(
15661568
condition=torch.device(torch_device).type == "cpu" and is_torch_version(">=", "2.5"),
15671569
reason="Test currently fails on CPU and PyTorch 2.5.1 but not on PyTorch 2.4.1.",

0 commit comments

Comments
 (0)