Skip to content

Commit a80b192

Browse files
authored
Support Flux Klein peft (fal) lora format (#13169)
peft (fal) lora format
1 parent 01de02e commit a80b192

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)