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

Support some z image lora formats. (#10978)

Browse Source
This commit is contained in:
comfyanonymous
2025-11-28 18:12:42 -08:00
committed by GitHub
parent b907085709
commit 52a32e2b32
2 changed files with 53 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
comfy/lora.py
View File

@@ -313,6 +313,14 @@ def model_lora_keys_unet(model, key_map={}):
key_map["transformer.{}".format(key_lora)] = k
key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = k #SimpleTuner lycoris format
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
return key_map