@@ -281,7 +281,8 @@ def lora_state_dict(
281281                The subfolder location of a model file within a larger model repository on the Hub or locally. 
282282            weight_name (`str`, *optional*, defaults to None): 
283283                Name of the serialized state dict file. 
284-             return_lora_metadata: TODO 
284+             return_lora_metadata (`bool`, *optional*, defaults to False): 
285+                 When enabled, additionally return the LoRA adapter metadata, typically found in the state dict. 
285286        """ 
286287        # Load the main state dict first which has the LoRA layers for either of 
287288        # UNet and text encoder or both. 
@@ -761,7 +762,8 @@ def lora_state_dict(
761762                The subfolder location of a model file within a larger model repository on the Hub or locally. 
762763            weight_name (`str`, *optional*, defaults to None): 
763764                Name of the serialized state dict file. 
764-             return_lora_metadata: TODO 
765+             return_lora_metadata (`bool`, *optional*, defaults to False): 
766+                 When enabled, additionally return the LoRA adapter metadata, typically found in the state dict. 
765767        """ 
766768        # Load the main state dict first which has the LoRA layers for either of 
767769        # UNet and text encoder or both. 
0 commit comments