updates

Author: sayakpaul

src/diffusers/loaders/lora_pipeline.py

@@ -842,9 +842,7 @@ def load_lora_into_unet(
         # If the serialization format is new (introduced in https://github.com/huggingface/diffusers/pull/2918),
         # then the `state_dict` keys should have `cls.unet_name` and/or `cls.text_encoder_name` as
         # their prefixes.
-        keys = list(state_dict.keys())
-        only_text_encoder = all(key.startswith(cls.text_encoder_name) for key in keys)
-        if not only_text_encoder:
+        if any(k.startswith(f"{cls.unet_name}.") for k in state_dict):
             # Load the layers corresponding to UNet.
             logger.info(f"Loading {cls.unet_name}.")
             unet.load_lora_adapter(
@@ -1008,6 +1006,11 @@ def load_lora_into_text_encoder(
                     _pipeline.enable_sequential_cpu_offload()
                 # Unsafe code />
 
+        else:
+            logger.info(
+                f"No LoRA keys found in the provided state dict for {text_encoder.__class__.__name__}. Please open an issue if you think this is unexpected - https://github.com/huggingface/diffusers/issues/new."
+            )
+
     @classmethod
     def save_lora_weights(
         cls,
@@ -1517,6 +1520,11 @@ def load_lora_into_text_encoder(
                     _pipeline.enable_sequential_cpu_offload()
                 # Unsafe code />
 
+        else:
+            logger.info(
+                f"No LoRA keys found in the provided state dict for {text_encoder.__class__.__name__}. Please open an issue if you think this is unexpected - https://github.com/huggingface/diffusers/issues/new."
+            )
+
     @classmethod
     def save_lora_weights(
         cls,
@@ -2146,6 +2154,11 @@ def load_lora_into_text_encoder(
                     _pipeline.enable_sequential_cpu_offload()
                 # Unsafe code />
 
+        else:
+            logger.info(
+                f"No LoRA keys found in the provided state dict for {text_encoder.__class__.__name__}. Please open an issue if you think this is unexpected - https://github.com/huggingface/diffusers/issues/new."
+            )
+
     @classmethod
     # Copied from diffusers.loaders.lora_pipeline.StableDiffusionLoraLoaderMixin.save_lora_weights with unet->transformer
     def save_lora_weights(
@@ -2580,6 +2593,11 @@ def load_lora_into_text_encoder(
                     _pipeline.enable_sequential_cpu_offload()
                 # Unsafe code />
 
+        else:
+            logger.info(
+                f"No LoRA keys found in the provided state dict for {text_encoder.__class__.__name__}. Please open an issue if you think this is unexpected - https://github.com/huggingface/diffusers/issues/new."
+            )
+
     @classmethod
     def save_lora_weights(
         cls,