_pack_dict_with_prefix

Author: sayakpaul

src/diffusers/loaders/lora_base.py

@@ -314,7 +314,7 @@ def _best_guess_weight_name(
     return weight_name
 
 
-def _pack_sd_with_prefix(state_dict, prefix):
+def _pack_dict_with_prefix(state_dict, prefix):
     sd_with_prefix = {f"{prefix}.{key}": value for key, value in state_dict.items()}
     return sd_with_prefix
 
@@ -914,7 +914,7 @@ def set_lora_device(self, adapter_names: List[str], device: Union[torch.device,
     @staticmethod
     def pack_weights(layers, prefix):
         layers_weights = layers.state_dict() if isinstance(layers, torch.nn.Module) else layers
-        return _pack_sd_with_prefix(layers_weights, prefix)
+        return _pack_dict_with_prefix(layers_weights, prefix)
 
     @staticmethod
     def write_lora_layers(

src/diffusers/loaders/lora_pipeline.py

@@ -37,7 +37,7 @@
     LoraBaseMixin,
     _fetch_state_dict,
     _load_lora_into_text_encoder,
-    _pack_sd_with_prefix,
+    _pack_dict_with_prefix,
 )
 from .lora_conversion_utils import (
     _convert_bfl_flux_control_lora_to_diffusers,
@@ -511,11 +511,11 @@ def save_lora_weights(
             state_dict.update(cls.pack_weights(text_encoder_lora_layers, cls.text_encoder_name))
 
         if unet_lora_adapter_metadata:
-            lora_adapter_metadata.update(_pack_sd_with_prefix(unet_lora_adapter_metadata, cls.unet_name))
+            lora_adapter_metadata.update(_pack_dict_with_prefix(unet_lora_adapter_metadata, cls.unet_name))
 
         if text_encoder_lora_adapter_metadata:
             lora_adapter_metadata.update(
-                _pack_sd_with_prefix(text_encoder_lora_adapter_metadata, cls.text_encoder_name)
+                _pack_dict_with_prefix(text_encoder_lora_adapter_metadata, cls.text_encoder_name)
             )
 
         # Save the model
@@ -2376,11 +2376,11 @@ def save_lora_weights(
             state_dict.update(cls.pack_weights(text_encoder_lora_layers, cls.text_encoder_name))
 
         if transformer_lora_adapter_metadata:
-            lora_adapter_metadata.update(_pack_sd_with_prefix(transformer_lora_adapter_metadata, cls.transformer_name))
+            lora_adapter_metadata.update(_pack_dict_with_prefix(transformer_lora_adapter_metadata, cls.transformer_name))
 
         if text_encoder_lora_adapter_metadata:
             lora_adapter_metadata.update(
-                _pack_sd_with_prefix(text_encoder_lora_adapter_metadata, cls.text_encoder_name)
+                _pack_dict_with_prefix(text_encoder_lora_adapter_metadata, cls.text_encoder_name)
             )
 
         # Save the model