Skip to content

Commit 170f644

Browse files
authored
Merge branch 'main' into transformers-v5-pr
2 parents e89b1b8 + a80b192 commit 170f644

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/diffusers/loaders/lora_pipeline.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5472,6 +5472,10 @@ def lora_state_dict(
54725472
logger.warning(warn_msg)
54735473
state_dict = {k: v for k, v in state_dict.items() if "dora_scale" not in k}
54745474

5475+
is_peft_format = any(k.startswith("base_model.model.") for k in state_dict)
5476+
if is_peft_format:
5477+
state_dict = {k.replace("base_model.model.", "diffusion_model."): v for k, v in state_dict.items()}
5478+
54755479
is_ai_toolkit = any(k.startswith("diffusion_model.") for k in state_dict)
54765480
if is_ai_toolkit:
54775481
state_dict = _convert_non_diffusers_flux2_lora_to_diffusers(state_dict)

0 commit comments

Comments
 (0)