We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7d0b9c4 commit c56add7Copy full SHA for c56add7
tests/lora/utils.py
@@ -32,6 +32,7 @@
32
from diffusers.utils.testing_utils import (
33
CaptureLogger,
34
floats_tensor,
35
+ is_torch_version,
36
require_peft_backend,
37
require_peft_version_greater,
38
require_transformers_version_greater,
@@ -1510,6 +1511,10 @@ def test_simple_inference_with_text_denoiser_multi_adapter_weighted(self):
1510
1511
)
1512
1513
@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
+ )
1518
def test_lora_fuse_nan(self):
1519
for scheduler_cls in self.scheduler_classes:
1520
components, text_lora_config, denoiser_lora_config = self.get_dummy_components(scheduler_cls)
0 commit comments