Skip to content

Commit 8e889c5

Browse files
authored
Support "transformer." LoRA prefix for Z-Image (#11135)
1 parent fd271de commit 8e889c5

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

comfy/lora.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -320,6 +320,7 @@ def model_lora_keys_unet(model, key_map={}):
320320
to = diffusers_keys[k]
321321
key_lora = k[:-len(".weight")]
322322
key_map["diffusion_model.{}".format(key_lora)] = to
323+
key_map["transformer.{}".format(key_lora)] = to
323324
key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = to
324325

325326
if isinstance(model, comfy.model_base.Kandinsky5):

0 commit comments

Comments
 (0)