diff --git a/tests/lora/test_lora_layers_peft.py b/tests/lora/test_lora_layers_peft.py index 1d8c697744..f6cd2a714a 100644 --- a/tests/lora/test_lora_layers_peft.py +++ b/tests/lora/test_lora_layers_peft.py @@ -1397,7 +1397,7 @@ class StableDiffusionXLLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase): @slow @require_torch_gpu -class LoraIntegrationTests(unittest.TestCase): +class LoraIntegrationTests(PeftLoraLoaderMixinTests, unittest.TestCase): def tearDown(self): import gc @@ -1650,7 +1650,7 @@ class LoraIntegrationTests(unittest.TestCase): @slow @require_torch_gpu -class LoraSDXLIntegrationTests(unittest.TestCase): +class LoraSDXLIntegrationTests(PeftLoraLoaderMixinTests, unittest.TestCase): def tearDown(self): import gc