Skip to content

Commit 2cc3683

Browse files
committed
fixes to fuse_lora, aryan was right.
1 parent 23854f2 commit 2cc3683

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

src/diffusers/loaders/lora_pipeline.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4243,10 +4243,9 @@ def save_lora_weights(
42434243
safe_serialization=safe_serialization,
42444244
)
42454245

4246-
# Copied from diffusers.loaders.lora_pipeline.Mochi1LoraLoaderMixin.fuse_lora
42474246
def fuse_lora(
42484247
self,
4249-
components: List[str] = ["transformer", "text_encoder"],
4248+
components: List[str] = ["transformer"],
42504249
lora_scale: float = 1.0,
42514250
safe_fusing: bool = False,
42524251
adapter_names: Optional[List[str]] = None,
@@ -4287,8 +4286,7 @@ def fuse_lora(
42874286
components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names
42884287
)
42894288

4290-
# Copied from diffusers.loaders.lora_pipeline.Mochi1LoraLoaderMixin.unfuse_lora
4291-
def unfuse_lora(self, components: List[str] = ["transformer", "text_encoder"], **kwargs):
4289+
def unfuse_lora(self, components: List[str] = ["transformer"], **kwargs):
42924290
r"""
42934291
Reverses the effect of
42944292
[`pipe.fuse_lora()`](https://huggingface.co/docs/diffusers/main/en/api/loaders#diffusers.loaders.LoraBaseMixin.fuse_lora).

tests/lora/test_lora_layers_hunyuanvideo.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -248,6 +248,7 @@ def test_original_format_cseti(self):
248248
out = out.flatten()
249249
out_slice = np.concatenate((out[:8], out[-8:]))
250250
from diffusers.utils.testing_utils import print_tensor_test
251+
251252
print_tensor_test(out_slice)
252253
# fmt: off
253254
expected_slice = np.array([0.1013, 0.1924, 0.0078, 0.1021, 0.1929, 0.0078, 0.1023, 0.1919, 0.7402, 0.104, 0.4482, 0.7354, 0.0925, 0.4382, 0.7275, 0.0815])

0 commit comments

Comments
 (0)