@@ -281,7 +281,8 @@ def lora_state_dict(
281
281
The subfolder location of a model file within a larger model repository on the Hub or locally.
282
282
weight_name (`str`, *optional*, defaults to None):
283
283
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.
285
286
"""
286
287
# Load the main state dict first which has the LoRA layers for either of
287
288
# UNet and text encoder or both.
@@ -761,7 +762,8 @@ def lora_state_dict(
761
762
The subfolder location of a model file within a larger model repository on the Hub or locally.
762
763
weight_name (`str`, *optional*, defaults to None):
763
764
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.
765
767
"""
766
768
# Load the main state dict first which has the LoRA layers for either of
767
769
# UNet and text encoder or both.
0 commit comments