diff --git a/tests/models/test_modeling_common.py b/tests/models/test_modeling_common.py index 57431d8b161b..a69ef872f54c 100644 --- a/tests/models/test_modeling_common.py +++ b/tests/models/test_modeling_common.py @@ -1061,7 +1061,7 @@ def test_deprecated_kwargs(self): @parameterized.expand([True, False]) @torch.no_grad() @unittest.skipIf(not is_peft_available(), "Only with PEFT") - def test_save_load_lora_adapter(self, use_dora=False): + def test_lora_save_load_adapter(self, use_dora=False): import safetensors from peft import LoraConfig from peft.utils import get_peft_model_state_dict @@ -1118,7 +1118,7 @@ def test_save_load_lora_adapter(self, use_dora=False): self.assertTrue(torch.allclose(outputs_with_lora, outputs_with_lora_2, atol=1e-4, rtol=1e-4)) @unittest.skipIf(not is_peft_available(), "Only with PEFT") - def test_wrong_adapter_name_raises_error(self): + def test_lora_wrong_adapter_name_raises_error(self): from peft import LoraConfig from diffusers.loaders.peft import PeftAdapterMixin