Skip to content

Commit 0eba7e7

Browse files
sayakpaulsayakpaul
committed
update
1 parent f78c6f9 commit 0eba7e7

File tree

1 file changed

+24
-12
lines changed

1 file changed

+24
-12
lines changed

‎src/diffusers/loaders/lora_pipeline.py

Lines changed: 24 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1168,7 +1168,8 @@ def lora_state_dict(
11681168
allowed by Git.
11691169
subfolder (`str`, *optional*, defaults to `""`):
11701170
The subfolder location of a model file within a larger model repository on the Hub or locally.
1171-
return_lora_metadata: TODO
1171+
return_lora_metadata (`bool`, *optional*, defaults to False):
1172+
When enabled, additionally return the LoRA adapter metadata, typically found in the state dict.
11721173
11731174
"""
11741175
# Load the main state dict first which has the LoRA layers for either of
@@ -1628,7 +1629,8 @@ def lora_state_dict(
16281629
allowed by Git.
16291630
subfolder (`str`, *optional*, defaults to `""`):
16301631
The subfolder location of a model file within a larger model repository on the Hub or locally.
1631-
return_lora_metadata: TODO
1632+
return_lora_metadata (`bool`, *optional*, defaults to False):
1633+
When enabled, additionally return the LoRA adapter metadata, typically found in the state dict.
16321634
16331635
"""
16341636
# Load the main state dict first which has the LoRA layers for either of
@@ -1972,7 +1974,8 @@ def lora_state_dict(
19721974
allowed by Git.
19731975
subfolder (`str`, *optional*, defaults to `""`):
19741976
The subfolder location of a model file within a larger model repository on the Hub or locally.
1975-
return_lora_metadata: TODO
1977+
return_lora_metadata (`bool`, *optional*, defaults to False):
1978+
When enabled, additionally return the LoRA adapter metadata, typically found in the state dict.
19761979
"""
19771980
# Load the main state dict first which has the LoRA layers for either of
19781981
# transformer and text encoder or both.
@@ -3006,7 +3009,8 @@ def lora_state_dict(
30063009
allowed by Git.
30073010
subfolder (`str`, *optional*, defaults to `""`):
30083011
The subfolder location of a model file within a larger model repository on the Hub or locally.
3009-
return_lora_metadata: TODO
3012+
return_lora_metadata (`bool`, *optional*, defaults to False):
3013+
When enabled, additionally return the LoRA adapter metadata, typically found in the state dict.
30103014
30113015
"""
30123016
# Load the main state dict first which has the LoRA layers for either of
@@ -3342,7 +3346,8 @@ def lora_state_dict(
33423346
allowed by Git.
33433347
subfolder (`str`, *optional*, defaults to `""`):
33443348
The subfolder location of a model file within a larger model repository on the Hub or locally.
3345-
return_lora_metadata: TODO
3349+
return_lora_metadata (`bool`, *optional*, defaults to False):
3350+
When enabled, additionally return the LoRA adapter metadata, typically found in the state dict.
33463351
33473352
"""
33483353
# Load the main state dict first which has the LoRA layers for either of
@@ -3680,7 +3685,8 @@ def lora_state_dict(
36803685
allowed by Git.
36813686
subfolder (`str`, *optional*, defaults to `""`):
36823687
The subfolder location of a model file within a larger model repository on the Hub or locally.
3683-
return_lora_metadata: TODO
3688+
return_lora_metadata (`bool`, *optional*, defaults to False):
3689+
When enabled, additionally return the LoRA adapter metadata, typically found in the state dict.
36843690
"""
36853691
# Load the main state dict first which has the LoRA layers for either of
36863692
# transformer and text encoder or both.
@@ -4022,7 +4028,8 @@ def lora_state_dict(
40224028
allowed by Git.
40234029
subfolder (`str`, *optional*, defaults to `""`):
40244030
The subfolder location of a model file within a larger model repository on the Hub or locally.
4025-
return_lora_metadata: TODO
4031+
return_lora_metadata (`bool`, *optional*, defaults to False):
4032+
When enabled, additionally return the LoRA adapter metadata, typically found in the state dict.
40264033
40274034
"""
40284035
# Load the main state dict first which has the LoRA layers for either of
@@ -4360,7 +4367,8 @@ def lora_state_dict(
43604367
allowed by Git.
43614368
subfolder (`str`, *optional*, defaults to `""`):
43624369
The subfolder location of a model file within a larger model repository on the Hub or locally.
4363-
return_lora_metadata: TODO
4370+
return_lora_metadata (`bool`, *optional*, defaults to False):
4371+
When enabled, additionally return the LoRA adapter metadata, typically found in the state dict.
43644372
"""
43654373
# Load the main state dict first which has the LoRA layers for either of
43664374
# transformer and text encoder or both.
@@ -4701,7 +4709,8 @@ def lora_state_dict(
47014709
allowed by Git.
47024710
subfolder (`str`, *optional*, defaults to `""`):
47034711
The subfolder location of a model file within a larger model repository on the Hub or locally.
4704-
return_lora_metadata: TODO
4712+
return_lora_metadata (`bool`, *optional*, defaults to False):
4713+
When enabled, additionally return the LoRA adapter metadata, typically found in the state dict.
47054714
"""
47064715
# Load the main state dict first which has the LoRA layers for either of
47074716
# transformer and text encoder or both.
@@ -5043,7 +5052,8 @@ def lora_state_dict(
50435052
allowed by Git.
50445053
subfolder (`str`, *optional*, defaults to `""`):
50455054
The subfolder location of a model file within a larger model repository on the Hub or locally.
5046-
return_lora_metadata: TODO
5055+
return_lora_metadata (`bool`, *optional*, defaults to False):
5056+
When enabled, additionally return the LoRA adapter metadata, typically found in the state dict.
50475057
"""
50485058
# Load the main state dict first which has the LoRA layers for either of
50495059
# transformer and text encoder or both.
@@ -5436,7 +5446,8 @@ def lora_state_dict(
54365446
allowed by Git.
54375447
subfolder (`str`, *optional*, defaults to `""`):
54385448
The subfolder location of a model file within a larger model repository on the Hub or locally.
5439-
return_lora_metadata: TODO
5449+
return_lora_metadata (`bool`, *optional*, defaults to False):
5450+
When enabled, additionally return the LoRA adapter metadata, typically found in the state dict.
54405451
54415452
"""
54425453
# Load the main state dict first which has the LoRA layers for either of
@@ -5774,7 +5785,8 @@ def lora_state_dict(
57745785
allowed by Git.
57755786
subfolder (`str`, *optional*, defaults to `""`):
57765787
The subfolder location of a model file within a larger model repository on the Hub or locally.
5777-
return_lora_metadata: TODO
5788+
return_lora_metadata (`bool`, *optional*, defaults to False):
5789+
When enabled, additionally return the LoRA adapter metadata, typically found in the state dict.
57785790
"""
57795791
# Load the main state dict first which has the LoRA layers for either of
57805792
# transformer and text encoder or both.

0 commit comments

Comments
 (0)