Skip to content

Commit cabcf3d

Browse files
committed
pr comments
1 parent bcb0924 commit cabcf3d

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/diffusers/loaders/lora_pipeline.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5271,6 +5271,11 @@ def load_lora_weights(
52715271

52725272
load_into_transformer_2 = kwargs.pop("load_into_transformer_2", False)
52735273
if load_into_transformer_2:
5274+
if geattr(self, "transformer_2", None) is None:
5275+
raise ValueError(
5276+
"Cannot load LoRA into transformer_2: transformer_2 is not available for this model"
5277+
"Ensure the model has a transformer_2 component before setting load_into_transformer_2=True."
5278+
)
52745279
self.load_lora_into_transformer(
52755280
state_dict,
52765281
transformer=self.transformer_2,

0 commit comments

Comments
 (0)