Skip to content

Commit 0c91c1a

Browse files
committed
Fix the repo consistency issues.
1 parent 147a356 commit 0c91c1a

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/diffusers/loaders/lora_pipeline.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1611,7 +1611,11 @@ def fuse_lora(
16111611
```
16121612
"""
16131613
super().fuse_lora(
1614-
components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names
1614+
components=components,
1615+
lora_scale=lora_scale,
1616+
safe_fusing=safe_fusing,
1617+
adapter_names=adapter_names,
1618+
**kwargs,
16151619
)
16161620

16171621
# Copied from diffusers.loaders.lora_pipeline.SanaLoraLoaderMixin.unfuse_lora
@@ -1630,7 +1634,7 @@ def unfuse_lora(self, components: List[str] = ["transformer", "text_encoder"], *
16301634
components (`List[str]`): List of LoRA-injectable components to unfuse LoRA from.
16311635
unfuse_transformer (`bool`, defaults to `True`): Whether to unfuse the UNet LoRA parameters.
16321636
"""
1633-
super().unfuse_lora(components=components)
1637+
super().unfuse_lora(components=components, **kwargs)
16341638

16351639

16361640
class FluxLoraLoaderMixin(LoraBaseMixin):

0 commit comments

Comments
 (0)