unify more.

Author: sayakpaul

tests/lora/test_lora_layers_auraflow.py

@@ -118,7 +118,3 @@ def test_modify_padding_mode(self):
     @unittest.skip("Text encoder LoRA is not supported in AuraFlow.")
     def test_simple_inference_with_partial_text_lora(self):
         pass
-
-    @unittest.skip("Text encoder LoRA is not supported in AuraFlow.")
-    def test_simple_inference_with_text_lora_and_scale(self):
-        pass

tests/lora/test_lora_layers_cogvideox.py

@@ -152,10 +152,6 @@ def test_modify_padding_mode(self):
     def test_simple_inference_with_partial_text_lora(self):
         pass
 
-    @unittest.skip("Text encoder LoRA is not supported in CogVideoX.")
-    def test_simple_inference_with_text_lora_and_scale(self):
-        pass
-
     @unittest.skip("Not supported in CogVideoX.")
     def test_simple_inference_with_text_denoiser_multi_adapter_block_lora(self):
         pass

tests/lora/test_lora_layers_cogview4.py

@@ -138,7 +138,3 @@ def test_modify_padding_mode(self):
     @unittest.skip("Text encoder LoRA is not supported in CogView4.")
     def test_simple_inference_with_partial_text_lora(self):
         pass
-
-    @unittest.skip("Text encoder LoRA is not supported in CogView4.")
-    def test_simple_inference_with_text_lora_and_scale(self):
-        pass

tests/lora/test_lora_layers_hunyuanvideo.py

@@ -172,10 +172,6 @@ def test_modify_padding_mode(self):
     def test_simple_inference_with_partial_text_lora(self):
         pass
 
-    @unittest.skip("Text encoder LoRA is not supported in HunyuanVideo.")
-    def test_simple_inference_with_text_lora_and_scale(self):
-        pass
-
 
 @nightly
 @require_torch_accelerator

tests/lora/test_lora_layers_ltx_video.py

@@ -129,7 +129,3 @@ def test_modify_padding_mode(self):
     @unittest.skip("Text encoder LoRA is not supported in LTXVideo.")
     def test_simple_inference_with_partial_text_lora(self):
         pass
-
-    @unittest.skip("Text encoder LoRA is not supported in LTXVideo.")
-    def test_simple_inference_with_text_lora_and_scale(self):
-        pass

tests/lora/test_lora_layers_lumina2.py

@@ -117,10 +117,6 @@ def test_modify_padding_mode(self):
     def test_simple_inference_with_partial_text_lora(self):
         pass
 
-    @unittest.skip("Text encoder LoRA is not supported in Lumina2.")
-    def test_simple_inference_with_text_lora_and_scale(self):
-        pass
-
     @unittest.skip("Text encoder LoRA is not supported in Lumina2.")
     @skip_mps
     @pytest.mark.xfail(

tests/lora/test_lora_layers_mochi.py

@@ -121,10 +121,6 @@ def test_modify_padding_mode(self):
     def test_simple_inference_with_partial_text_lora(self):
         pass
 
-    @unittest.skip("Text encoder LoRA is not supported in Mochi.")
-    def test_simple_inference_with_text_lora_and_scale(self):
-        pass
-
     @unittest.skip("Not supported in Mochi.")
     def test_simple_inference_with_text_denoiser_multi_adapter_block_lora(self):
         pass

tests/lora/test_lora_layers_sana.py

@@ -120,7 +120,3 @@ def test_simple_inference_with_text_denoiser_block_scale_for_all_dict_options(se
     @unittest.skip("Text encoder LoRA is not supported in SANA.")
     def test_simple_inference_with_partial_text_lora(self):
         pass
-
-    @unittest.skip("Text encoder LoRA is not supported in SANA.")
-    def test_simple_inference_with_text_lora_and_scale(self):
-        pass

tests/lora/test_lora_layers_wan.py

@@ -125,7 +125,3 @@ def test_modify_padding_mode(self):
     @unittest.skip("Text encoder LoRA is not supported in Wan.")
     def test_simple_inference_with_partial_text_lora(self):
         pass
-
-    @unittest.skip("Text encoder LoRA is not supported in Wan.")
-    def test_simple_inference_with_text_lora_and_scale(self):
-        pass

tests/lora/utils.py

@@ -391,12 +391,7 @@ def test_lora_text_encoder_actions(self, action):
                         "Loading from a saved checkpoint should yield the same result as the original LoRA.",
                     )
 
-    @parameterized.expand(
-        [
-            ("text_encoder_only",),
-            ("text_and_denoiser",),
-        ]
-    )
+    @parameterized.expand([("text_encoder_only",), ("text_and_denoiser",)])
     def test_lora_scaling(self, lora_components_to_add):
         """
         Tests inference with LoRA scaling applied via attention_kwargs