Skip to content

Commit 59d1caa

Browse files
authored
Remove peft tests from old lora backend tests (#6273)
update
1 parent c022e52 commit 59d1caa

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/lora/test_lora_layers_peft.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1397,7 +1397,7 @@ class StableDiffusionXLLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase):
13971397

13981398
@slow
13991399
@require_torch_gpu
1400-
class LoraIntegrationTests(unittest.TestCase):
1400+
class LoraIntegrationTests(PeftLoraLoaderMixinTests, unittest.TestCase):
14011401
def tearDown(self):
14021402
import gc
14031403

@@ -1650,7 +1650,7 @@ def test_load_unload_load_kohya_lora(self):
16501650

16511651
@slow
16521652
@require_torch_gpu
1653-
class LoraSDXLIntegrationTests(unittest.TestCase):
1653+
class LoraSDXLIntegrationTests(PeftLoraLoaderMixinTests, unittest.TestCase):
16541654
def tearDown(self):
16551655
import gc
16561656

0 commit comments

Comments
 (0)