Skip to content

Commit eb52469

Browse files
committed
TODO 1.
1 parent 9bbc6dc commit eb52469

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/diffusers/loaders/lora_pipeline.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)