Skip to content

Commit ae42182

Browse files
fix: detect Z-Image LoRAs with transformer.layers prefix (#8986)
OneTrainer exports Z-Image LoRAs with 'transformer.layers.' key prefix instead of 'diffusion_model.layers.'. Add this prefix (and the PEFT-wrapped 'base_model.model.transformer.layers.' variant) to the Z-Image LoRA probe so these models are correctly identified and loaded.
1 parent 32002bd commit ae42182

File tree

1 file changed

+4
-0
lines changed
  • invokeai/backend/model_manager/configs

1 file changed

+4
-0
lines changed

invokeai/backend/model_manager/configs/lora.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -711,6 +711,8 @@ def _validate_looks_like_lora(cls, mod: ModelOnDisk) -> None:
711711
state_dict,
712712
{
713713
"diffusion_model.layers.", # Z-Image S3-DiT layer pattern
714+
"transformer.layers.", # OneTrainer/diffusers prefix variant
715+
"base_model.model.transformer.layers.", # PEFT-wrapped variant
714716
},
715717
)
716718

@@ -747,6 +749,8 @@ def _get_base_or_raise(cls, mod: ModelOnDisk) -> BaseModelType:
747749
state_dict,
748750
{
749751
"diffusion_model.layers.", # Z-Image S3-DiT layer pattern
752+
"transformer.layers.", # OneTrainer/diffusers prefix variant
753+
"base_model.model.transformer.layers.", # PEFT-wrapped variant
750754
},
751755
)
752756

0 commit comments

Comments
 (0)