We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents e89b1b8 + a80b192 commit 170f644Copy full SHA for 170f644
src/diffusers/loaders/lora_pipeline.py
@@ -5472,6 +5472,10 @@ def lora_state_dict(
5472
logger.warning(warn_msg)
5473
state_dict = {k: v for k, v in state_dict.items() if "dora_scale" not in k}
5474
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
+
5479
is_ai_toolkit = any(k.startswith("diffusion_model.") for k in state_dict)
5480
if is_ai_toolkit:
5481
state_dict = _convert_non_diffusers_flux2_lora_to_diffusers(state_dict)
0 commit comments