add copied.

sayakpaul · sayakpaul · commit 5fbc59c7b2cc · 2025-01-07T12:27:37.000+05:30
diff --git a/src/diffusers/loaders/lora_pipeline.py b/src/diffusers/loaders/lora_pipeline.py
@@ -4243,6 +4243,7 @@ def save_lora_weights(
             safe_serialization=safe_serialization,
         )
 
+    # Copied from diffusers.loaders.lora_pipeline.CogVideoXLoraLoaderMixin.fuse_lora
     def fuse_lora(
         self,
         components: List[str] = ["transformer"],
@@ -4288,7 +4289,7 @@ def fuse_lora(
             components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names
         )
 
-    # Copied from diffusers.loaders.lora_pipeline.StableDiffusionLoraLoaderMixin.unfuse_lora with unet->transformer
+    # Copied from diffusers.loaders.lora_pipeline.CogVideoXLoraLoaderMixin.unfuse_lora
     def unfuse_lora(self, components: List[str] = ["transformer", "text_encoder"], **kwargs):
         r"""
         Reverses the effect of

Original file line number	Diff line number	Diff line change
`@@ -4243,6 +4243,7 @@ def save_lora_weights(`
`4243`	`4243`	`safe_serialization=safe_serialization,`
`4244`	`4244`	`)`
`4245`	`4245`
	`4246`	`+ # Copied from diffusers.loaders.lora_pipeline.CogVideoXLoraLoaderMixin.fuse_lora`
`4246`	`4247`	`def fuse_lora(`
`4247`	`4248`	`self,`
`4248`	`4249`	`components: List[str] = ["transformer"],`
`@@ -4288,7 +4289,7 @@ def fuse_lora(`
`4288`	`4289`	`components=components, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names`
`4289`	`4290`	`)`
`4290`	`4291`
`4291`		`- # Copied from diffusers.loaders.lora_pipeline.StableDiffusionLoraLoaderMixin.unfuse_lora with unet->transformer`
	`4292`	`+ # Copied from diffusers.loaders.lora_pipeline.CogVideoXLoraLoaderMixin.unfuse_lora`
`4292`	`4293`	`def unfuse_lora(self, components: List[str] = ["transformer", "text_encoder"], **kwargs):`
`4293`	`4294`	`r"""`
`4294`	`4295`	`Reverses the effect of`