Remove peft tests from old lora backend tests (#6273)

update

Remove peft tests from old lora backend tests (#6273)
update
59d1caa2 · Dhruv Nair · GitHub · c022e529 · 59d1caa2
Unverified Commit 59d1caa2 authored Dec 22, 2023 by Dhruv Nair Committed by GitHub Dec 22, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/lora/test_lora_layers_peft.py tests/lora/test_lora_layers_peft.py +2 -2

No files found.
--- 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