Merge 51119d3283056817b6260107f535ae356df2ee62 into 56fa7dbe380cb5591c5542f8aa51ce2fc26beedf

Support "transformer." LoRA prefix for Z-Image
Please try to be consistent in your LoRA loading code Qwen supports the "transformer." prefix. This PR adds it to Z-Image (for some reason the code for Lumina2 is used for Z-Image)
2025-12-08 21:44:33 +08:00 · 2025-12-07 09:42:26 -05:00 · 2025-12-06 01:28:18 +01:00
1 changed files with 1 additions and 0 deletions
--- a/comfy/lora.py
+++ b/comfy/lora.py
@ -320,6 +320,7 @@ def model_lora_keys_unet(model, key_map={}):
                to = diffusers_keys[k]
                key_lora = k[:-len(".weight")]
                key_map["diffusion_model.{}".format(key_lora)] = to
+                key_map["transformer.{}".format(key_lora)] = to
                key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = to

    if isinstance(model, comfy.model_base.Kandinsky5):
Author	SHA1	Message	Date
dxqb	180e7738f7	Merge 51119d3283056817b6260107f535ae356df2ee62 into 56fa7dbe380cb5591c5542f8aa51ce2fc26beedf	2025-12-07 09:42:26 -05:00
dxqb	51119d3283	Support "transformer." LoRA prefix for Z-Image Please try to be consistent in your LoRA loading code Qwen supports the "transformer." prefix. This PR adds it to Z-Image (for some reason the code for Lumina2 is used for Z-Image)	2025-12-06 01:28:18 +01:00