cog

Author: sayakpaul

tests/lora/test_lora_layers_cogvideox.py

@@ -29,6 +29,7 @@
 from diffusers.utils.testing_utils import (
     floats_tensor,
     is_peft_available,
+    is_torch_version,
     require_peft_backend,
     skip_mps,
     torch_device,
@@ -126,6 +127,10 @@ def get_dummy_inputs(self, with_generator=True):
         return noise, input_ids, pipeline_inputs
 
     @skip_mps
+    @unittest.skipIf(
+        torch.device(torch_device).type == "cpu" and is_torch_version(">=", "2.5"),
+        "Test not supported on PyTorch 2.5 and CPU.",
+    )
     def test_lora_fuse_nan(self):
         for scheduler_cls in self.scheduler_classes:
             components, text_lora_config, denoiser_lora_config = self.get_dummy_components(scheduler_cls)