Skip to content

Commit c56add7

Browse files
committed
skip nan lora tests on PyTorch 2.5.1 CPU.
1 parent 7d0b9c4 commit c56add7

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/lora/utils.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
from diffusers.utils.testing_utils import (
3333
CaptureLogger,
3434
floats_tensor,
35+
is_torch_version,
3536
require_peft_backend,
3637
require_peft_version_greater,
3738
require_transformers_version_greater,
@@ -1510,6 +1511,10 @@ def test_simple_inference_with_text_denoiser_multi_adapter_weighted(self):
15101511
)
15111512

15121513
@skip_mps
1514+
@unittest.skipIf(
1515+
torch.device(torch_device).type == "cpu" and is_torch_version(">=", "2.5"),
1516+
"Test not supported on PyTorch 2.5 and CPU.",
1517+
)
15131518
def test_lora_fuse_nan(self):
15141519
for scheduler_cls in self.scheduler_classes:
15151520
components, text_lora_config, denoiser_lora_config = self.get_dummy_components(scheduler_cls)

0 commit comments

Comments
 (0)