Skip to content

Commit d5f6fa5

Browse files
Apply style fixes
1 parent b78bac4 commit d5f6fa5

File tree

1 file changed

+55
-11
lines changed

1 file changed

+55
-11
lines changed

src/diffusers/loaders/lora_pipeline.py

Lines changed: 55 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -451,7 +451,11 @@ def fuse_lora(
451451
```
452452
"""
453453
super().fuse_lora(
454-
components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names, **kwargs
454+
components=components,
455+
lora_scale=lora_scale,
456+
safe_fusing=safe_fusing,
457+
adapter_names=adapter_names,
458+
**kwargs,
455459
)
456460

457461
def unfuse_lora(self, components: List[str] = ["unet", "text_encoder"], **kwargs):
@@ -891,7 +895,11 @@ def fuse_lora(
891895
```
892896
"""
893897
super().fuse_lora(
894-
components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names, **kwargs
898+
components=components,
899+
lora_scale=lora_scale,
900+
safe_fusing=safe_fusing,
901+
adapter_names=adapter_names,
902+
**kwargs,
895903
)
896904

897905
def unfuse_lora(self, components: List[str] = ["unet", "text_encoder", "text_encoder_2"], **kwargs):
@@ -1290,7 +1298,11 @@ def fuse_lora(
12901298
```
12911299
"""
12921300
super().fuse_lora(
1293-
components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names, **kwargs
1301+
components=components,
1302+
lora_scale=lora_scale,
1303+
safe_fusing=safe_fusing,
1304+
adapter_names=adapter_names,
1305+
**kwargs,
12941306
)
12951307

12961308
# Copied from diffusers.loaders.lora_pipeline.StableDiffusionXLLoraLoaderMixin.unfuse_lora with unet->transformer
@@ -1828,7 +1840,11 @@ def fuse_lora(
18281840
)
18291841

18301842
super().fuse_lora(
1831-
components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names, **kwargs
1843+
components=components,
1844+
lora_scale=lora_scale,
1845+
safe_fusing=safe_fusing,
1846+
adapter_names=adapter_names,
1847+
**kwargs,
18321848
)
18331849

18341850
def unfuse_lora(self, components: List[str] = ["transformer", "text_encoder"], **kwargs):
@@ -2548,7 +2564,11 @@ def fuse_lora(
25482564
```
25492565
"""
25502566
super().fuse_lora(
2551-
components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names, **kwargs
2567+
components=components,
2568+
lora_scale=lora_scale,
2569+
safe_fusing=safe_fusing,
2570+
adapter_names=adapter_names,
2571+
**kwargs,
25522572
)
25532573

25542574
def unfuse_lora(self, components: List[str] = ["transformer"], **kwargs):
@@ -2852,7 +2872,11 @@ def fuse_lora(
28522872
```
28532873
"""
28542874
super().fuse_lora(
2855-
components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names, **kwargs
2875+
components=components,
2876+
lora_scale=lora_scale,
2877+
safe_fusing=safe_fusing,
2878+
adapter_names=adapter_names,
2879+
**kwargs,
28562880
)
28572881

28582882
# Copied from diffusers.loaders.lora_pipeline.CogVideoXLoraLoaderMixin.unfuse_lora
@@ -3157,7 +3181,11 @@ def fuse_lora(
31573181
```
31583182
"""
31593183
super().fuse_lora(
3160-
components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names, **kwargs
3184+
components=components,
3185+
lora_scale=lora_scale,
3186+
safe_fusing=safe_fusing,
3187+
adapter_names=adapter_names,
3188+
**kwargs,
31613189
)
31623190

31633191
# Copied from diffusers.loaders.lora_pipeline.CogVideoXLoraLoaderMixin.unfuse_lora
@@ -3462,7 +3490,11 @@ def fuse_lora(
34623490
```
34633491
"""
34643492
super().fuse_lora(
3465-
components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names, **kwargs
3493+
components=components,
3494+
lora_scale=lora_scale,
3495+
safe_fusing=safe_fusing,
3496+
adapter_names=adapter_names,
3497+
**kwargs,
34663498
)
34673499

34683500
# Copied from diffusers.loaders.lora_pipeline.CogVideoXLoraLoaderMixin.unfuse_lora
@@ -3770,7 +3802,11 @@ def fuse_lora(
37703802
```
37713803
"""
37723804
super().fuse_lora(
3773-
components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names, **kwargs
3805+
components=components,
3806+
lora_scale=lora_scale,
3807+
safe_fusing=safe_fusing,
3808+
adapter_names=adapter_names,
3809+
**kwargs,
37743810
)
37753811

37763812
# Copied from diffusers.loaders.lora_pipeline.CogVideoXLoraLoaderMixin.unfuse_lora
@@ -4079,7 +4115,11 @@ def fuse_lora(
40794115
```
40804116
"""
40814117
super().fuse_lora(
4082-
components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names, **kwargs
4118+
components=components,
4119+
lora_scale=lora_scale,
4120+
safe_fusing=safe_fusing,
4121+
adapter_names=adapter_names,
4122+
**kwargs,
40834123
)
40844124

40854125
# Copied from diffusers.loaders.lora_pipeline.SanaLoraLoaderMixin.unfuse_lora
@@ -4384,7 +4424,11 @@ def fuse_lora(
43844424
```
43854425
"""
43864426
super().fuse_lora(
4387-
components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names, **kwargs
4427+
components=components,
4428+
lora_scale=lora_scale,
4429+
safe_fusing=safe_fusing,
4430+
adapter_names=adapter_names,
4431+
**kwargs,
43884432
)
43894433

43904434
# Copied from diffusers.loaders.lora_pipeline.CogVideoXLoraLoaderMixin.unfuse_lora

0 commit comments

Comments
 (0)