Update tests/lora/test_lora_layers_flux.py

Co-authored-by: Sayak Paul <[email protected]>

Author: hlky

tests/lora/test_lora_layers_flux.py

@@ -195,10 +195,10 @@ def test_lora_expansion_works_for_absent_keys(self):
             # Modify the state dict to exclude "x_embedder" related LoRA params.
             lora_state_dict = safetensors.torch.load_file(os.path.join(tmpdirname, "pytorch_lora_weights.safetensors"))
             lora_state_dict_without_xembedder = {k: v for k, v in lora_state_dict.items() if "x_embedder" not in k}
-            pipe.load_lora_weights(lora_state_dict_without_xembedder, adapter_name="two")
+            pipe.load_lora_weights(lora_state_dict_without_xembedder, adapter_name="one")
 
             # Load state dict with `x_embedder`.
-            pipe.load_lora_weights(os.path.join(tmpdirname, "pytorch_lora_weights.safetensors"), adapter_name="one")
+            pipe.load_lora_weights(os.path.join(tmpdirname, "pytorch_lora_weights.safetensors"), adapter_name="two")
 
         pipe.set_adapters(["one", "two"])
         self.assertTrue(check_if_lora_correctly_set(pipe.transformer), "Lora not correctly set in transformer")