up

Author: sayakpaul

tests/lora/test_lora_layers_auraflow.py

@@ -40,7 +40,7 @@
 
 
 @require_peft_backend
-class AuraFlowLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
+class AuraFlowLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase):
     pipeline_class = AuraFlowPipeline
     scheduler_cls = FlowMatchEulerDiscreteScheduler
     scheduler_classes = [FlowMatchEulerDiscreteScheduler]

tests/lora/test_lora_layers_cogvideox.py

@@ -40,7 +40,7 @@
 
 
 @require_peft_backend
-class CogVideoXLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
+class CogVideoXLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase):
     pipeline_class = CogVideoXPipeline
     scheduler_cls = CogVideoXDPMScheduler
     scheduler_kwargs = {"timestep_spacing": "trailing"}

tests/lora/test_lora_layers_cogview4.py

@@ -47,7 +47,7 @@ def from_pretrained(*args, **kwargs):
 
 @require_peft_backend
 @skip_mps
-class CogView4LoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
+class CogView4LoRATests(PeftLoraLoaderMixinTests, unittest.TestCase):
     pipeline_class = CogView4Pipeline
     scheduler_cls = FlowMatchEulerDiscreteScheduler
     scheduler_classes = [FlowMatchEulerDiscreteScheduler]

tests/lora/test_lora_layers_flux.py

@@ -53,7 +53,7 @@
 
 
 @require_peft_backend
-class FluxLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
+class FluxLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase):
     pipeline_class = FluxPipeline
     scheduler_cls = FlowMatchEulerDiscreteScheduler()
     scheduler_kwargs = {}
@@ -280,7 +280,7 @@ def test_simple_inference_with_text_denoiser_multi_adapter_block_lora(self):
         pass
 
 
-class FluxControlLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
+class FluxControlLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase):
     pipeline_class = FluxControlPipeline
     scheduler_cls = FlowMatchEulerDiscreteScheduler()
     scheduler_kwargs = {}

tests/lora/test_lora_layers_hunyuanvideo.py

@@ -48,7 +48,7 @@
 
 @require_peft_backend
 @skip_mps
-class HunyuanVideoLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
+class HunyuanVideoLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase):
     pipeline_class = HunyuanVideoPipeline
     scheduler_cls = FlowMatchEulerDiscreteScheduler
     scheduler_classes = [FlowMatchEulerDiscreteScheduler]

tests/lora/test_lora_layers_ltx_video.py

@@ -34,7 +34,7 @@
 
 
 @require_peft_backend
-class LTXVideoLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
+class LTXVideoLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase):
     pipeline_class = LTXPipeline
     scheduler_cls = FlowMatchEulerDiscreteScheduler
     scheduler_classes = [FlowMatchEulerDiscreteScheduler]

tests/lora/test_lora_layers_mochi.py

@@ -34,7 +34,7 @@
 
 @require_peft_backend
 @skip_mps
-class MochiLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
+class MochiLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase):
     pipeline_class = MochiPipeline
     scheduler_cls = FlowMatchEulerDiscreteScheduler
     scheduler_classes = [FlowMatchEulerDiscreteScheduler]

tests/lora/test_lora_layers_qwenimage.py

@@ -34,7 +34,7 @@
 
 
 @require_peft_backend
-class QwenImageLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
+class QwenImageLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase):
     pipeline_class = QwenImagePipeline
     scheduler_cls = FlowMatchEulerDiscreteScheduler
     scheduler_classes = [FlowMatchEulerDiscreteScheduler]

tests/lora/test_lora_layers_sana.py

@@ -29,7 +29,7 @@
 
 
 @require_peft_backend
-class SanaLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
+class SanaLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase):
     pipeline_class = SanaPipeline
     scheduler_cls = FlowMatchEulerDiscreteScheduler(shift=7.0)
     scheduler_kwargs = {}

tests/lora/test_lora_layers_sd3.py

@@ -51,7 +51,7 @@
 
 
 @require_peft_backend
-class SD3LoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
+class SD3LoRATests(PeftLoraLoaderMixinTests, unittest.TestCase):
     pipeline_class = StableDiffusion3Pipeline
     scheduler_cls = FlowMatchEulerDiscreteScheduler
     scheduler_kwargs = {}