Unverified Commit 59d1caa2 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

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

update
parent c022e529
...@@ -1397,7 +1397,7 @@ class StableDiffusionXLLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase): ...@@ -1397,7 +1397,7 @@ class StableDiffusionXLLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase):
@slow @slow
@require_torch_gpu @require_torch_gpu
class LoraIntegrationTests(unittest.TestCase): class LoraIntegrationTests(PeftLoraLoaderMixinTests, unittest.TestCase):
def tearDown(self): def tearDown(self):
import gc import gc
...@@ -1650,7 +1650,7 @@ class LoraIntegrationTests(unittest.TestCase): ...@@ -1650,7 +1650,7 @@ class LoraIntegrationTests(unittest.TestCase):
@slow @slow
@require_torch_gpu @require_torch_gpu
class LoraSDXLIntegrationTests(unittest.TestCase): class LoraSDXLIntegrationTests(PeftLoraLoaderMixinTests, unittest.TestCase):
def tearDown(self): def tearDown(self):
import gc import gc
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment