Skip to content

Commit 4c425e2

Browse files
author
linoy
committed
fix copies
1 parent 18382f4 commit 4c425e2

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/diffusers/loaders/lora_pipeline.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5693,10 +5693,13 @@ def load_lora_weights(
56935693
load_into_transformer_2 = kwargs.pop("load_into_transformer_2", False)
56945694
if load_into_transformer_2:
56955695
if not hasattr(self, "transformer_2"):
5696-
raise ValueError(
5697-
"Cannot load LoRA into transformer_2: transformer_2 is not available for this model"
5696+
raise AttributeError(
5697+
f"'{type(self).__name__}' object has no attribute transformer_2"
5698+
"Note that Wan2.1 models do not have a transformer_2 component."
56985699
"Ensure the model has a transformer_2 component before setting load_into_transformer_2=True."
56995700
)
5701+
if "transformer_2" not in self._lora_loadable_modules:
5702+
self._lora_loadable_modules.append("transformer_2")
57005703
self.load_lora_into_transformer(
57015704
state_dict,
57025705
transformer=self.transformer_2,

0 commit comments

Comments
 (0)