Support "transformer." LoRA prefix for Z-Image (#11135)

This commit is contained in:
dxqb 2025-12-08 21:17:26 +01:00 committed by GitHub
parent fd271dedfd
commit 8e889c535d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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):