diff --git a/tests/lora/test_lora_layers_sana.py b/tests/lora/test_lora_layers_sana.py index a08908c610..a7d0261ce0 100644 --- a/tests/lora/test_lora_layers_sana.py +++ b/tests/lora/test_lora_layers_sana.py @@ -29,7 +29,7 @@ from .utils import PeftLoraLoaderMixinTests # noqa: E402 @require_peft_backend -class SanaLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests): +class SanaLoRATests(PeftLoraLoaderMixinTests, unittest.TestCase): pipeline_class = SanaPipeline scheduler_cls = FlowMatchEulerDiscreteScheduler(shift=7.0) scheduler_kwargs = {}