jaidaken/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add some missing z image lora layers. (#10980)

Browse Source
This commit is contained in:
comfyanonymous
2025-11-28 20:55:00 -08:00
committed by GitHub
parent af96d9812d
commit 5151cff293
2 changed files with 41 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
comfy/lora.py
View File

@@ -316,10 +316,11 @@ def model_lora_keys_unet(model, key_map={}):
if isinstance(model, comfy.model_base.Lumina2):
diffusers_keys = comfy.utils.z_image_to_diffusers(model.model_config.unet_config, output_prefix="diffusion_model.")
for k in diffusers_keys:
to = diffusers_keys[k]
key_lora = k[:-len(".weight")]
key_map["diffusion_model.{}".format(key_lora)] = to
key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = to
if k.endswith(".weight"):
to = diffusers_keys[k]
key_lora = k[:-len(".weight")]
key_map["diffusion_model.{}".format(key_lora)] = to
key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = to
return key_map