smol updates

Author: sayakpaul

src/diffusers/loaders/peft.py

@@ -239,11 +239,11 @@ def load_lora_adapter(
             raise ValueError("`network_alphas` cannot be None when `prefix` is None.")
 
         if prefix is not None:
-            metadata = state_dict.pop("lora_metadata", None)
+            metadata = state_dict.pop("lora_adapter_metadata", None)
             state_dict = {k[len(f"{prefix}.") :]: v for k, v in state_dict.items() if k.startswith(f"{prefix}.")}
 
             if metadata is not None:
-                state_dict["lora_metadata"] = metadata
+                state_dict["lora_adapter_metadata"] = metadata
 
         if len(state_dict) > 0:
             if adapter_name in getattr(self, "peft_config", {}) and not hotswap:

src/diffusers/utils/peft_utils.py

@@ -151,9 +151,9 @@ def get_peft_kwargs(
     rank_dict, network_alpha_dict, peft_state_dict, is_unet=True, prefix=None, load_with_metadata=False
 ):
     if load_with_metadata:
-        if "lora_metadata" not in peft_state_dict:
-            raise ValueError("Couldn't find 'lora_metadata' key in the `peft_state_dict`.")
-        metadata = peft_state_dict["lora_metadata"]
+        if "lora_adapter_metadata" not in peft_state_dict:
+            raise ValueError("Couldn't find 'lora_adapter_metadata' key in the `peft_state_dict`.")
+        metadata = peft_state_dict["lora_adapter_metadata"]
         if prefix is not None:
             metadata = {k.replace(f"{prefix}.", ""): v for k, v in metadata.items()}
         return metadata

src/diffusers/utils/state_dict_utils.py

@@ -360,7 +360,7 @@ def _maybe_populate_state_dict_with_metadata(state_dict, model_file, metadata_ke
                 metadata_keys = list(metadata.keys())
                 if not (len(metadata_keys) == 1 and metadata_keys[0] == "format"):
                     peft_metadata = {k: v for k, v in metadata.items() if k != "format"}
-                    state_dict["lora_metadata"] = json.loads(peft_metadata[metadata_key])
+                    state_dict["lora_adapter_metadata"] = json.loads(peft_metadata[metadata_key])
         else:
             raise ValueError("Metadata couldn't be parsed from the safetensors file.")
     return state_dict