git/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2026-04-18 04:45:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: detect Z-Image LoRAs with transformer.layers prefix (#8986)

Browse Source 
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.
This commit is contained in:
Alexander Eichhorn 2026-04-07 03:52:06 +02:00 committed by GitHub
parent 32002bd37e
commit ae42182246
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/model_manager/configs/lora.py
View File

@ -711,6 +711,8 @@ class LoRA_LyCORIS_ZImage_Config(LoRA_LyCORIS_Config_Base, Config_Base):
state_dict,
{
"diffusion_model.layers.", # Z-Image S3-DiT layer pattern
"transformer.layers.", # OneTrainer/diffusers prefix variant
"base_model.model.transformer.layers.", # PEFT-wrapped variant
},
)
@ -747,6 +749,8 @@ class LoRA_LyCORIS_ZImage_Config(LoRA_LyCORIS_Config_Base, Config_Base):
state_dict,
{
"diffusion_model.layers.", # Z-Image S3-DiT layer pattern
"transformer.layers.", # OneTrainer/diffusers prefix variant
"base_model.model.transformer.layers.", # PEFT-wrapped variant
},
)