Skip to content

Commit 6134491

Browse files
committed
updates
1 parent 1db7503 commit 6134491

File tree

1 file changed

+7
-11
lines changed

1 file changed

+7
-11
lines changed

src/diffusers/loaders/lora_pipeline.py

Lines changed: 7 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -2401,17 +2401,13 @@ def load_lora_into_transformer(
24012401
)
24022402

24032403
# Load the layers corresponding to transformer.
2404-
keys = list(state_dict.keys())
2405-
transformer_present = any(key.startswith(cls.transformer_name) for key in keys)
2406-
if transformer_present:
2407-
logger.info(f"Loading {cls.transformer_name}.")
2408-
transformer.load_lora_adapter(
2409-
state_dict,
2410-
network_alphas=network_alphas,
2411-
adapter_name=adapter_name,
2412-
_pipeline=_pipeline,
2413-
low_cpu_mem_usage=low_cpu_mem_usage,
2414-
)
2404+
transformer.load_lora_adapter(
2405+
state_dict,
2406+
network_alphas=network_alphas,
2407+
adapter_name=adapter_name,
2408+
_pipeline=_pipeline,
2409+
low_cpu_mem_usage=low_cpu_mem_usage,
2410+
)
24152411

24162412
@classmethod
24172413
# Copied from diffusers.loaders.lora_pipeline.StableDiffusionLoraLoaderMixin.load_lora_into_text_encoder

0 commit comments

Comments
 (0)