Skip test_lora_fuse_nan on mps (#7481)

UmerHA · sayakpaul · web-flow · commit 0b8e29289dd9 · 2024-03-27T14:35:59.000+05:30
Skipping test_lora_fuse_nan on mps

Co-authored-by: Sayak Paul &lt;spsayakpaul@gmail.com&gt;
diff --git a/tests/lora/utils.py b/tests/lora/utils.py
@@ -31,6 +31,7 @@
     floats_tensor,
     require_peft_backend,
     require_peft_version_greater,
+    skip_mps,
     torch_device,
 )
 
@@ -923,6 +924,7 @@ def test_simple_inference_with_text_unet_multi_adapter_weighted(self):
                 "output with no lora and output with lora disabled should give same results",
             )
 
+    @skip_mps
     def test_lora_fuse_nan(self):
         for scheduler_cls in [DDIMScheduler, LCMScheduler]:
             components, text_lora_config, unet_lora_config = self.get_dummy_components(scheduler_cls)

Original file line number	Diff line number	Diff line change
`@@ -31,6 +31,7 @@`
`31`	`31`	`floats_tensor,`
`32`	`32`	`require_peft_backend,`
`33`	`33`	`require_peft_version_greater,`
	`34`	`+ skip_mps,`
`34`	`35`	`torch_device,`
`35`	`36`	`)`
`36`	`37`
`@@ -923,6 +924,7 @@ def test_simple_inference_with_text_unet_multi_adapter_weighted(self):`
`923`	`924`	`"output with no lora and output with lora disabled should give same results",`
`924`	`925`	`)`
`925`	`926`
	`927`	`+ @skip_mps`
`926`	`928`	`def test_lora_fuse_nan(self):`
`927`	`929`	`for scheduler_cls in [DDIMScheduler, LCMScheduler]:`
`928`	`930`	`components, text_lora_config, unet_lora_config = self.get_dummy_components(scheduler_cls)`