wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-03 13:19:23 +08:00
Code Issues Actions 21 Packages Projects Releases Wiki Activity

support diffusers loras

Browse Source
This commit is contained in:
kijai 2026-06-22 22:16:16 +03:00
parent 08381b1a9d
commit c2a95725c0
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
comfy/lora.py
View File

@ -326,6 +326,34 @@ 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.Krea2):
krea2_rename = {
"first": "img_in", "tmlp.0": "time_embed.linear_1", "tmlp.2": "time_embed.linear_2",
"tproj.1": "time_mod_proj", "txtmlp.1": "txt_in.linear_1", "txtmlp.3": "txt_in.linear_2",
"last.linear": "final_layer.linear",
}
krea2_sub = {
"txtfusion.": "text_fusion.", ".mlp.": ".ff.",
".attn.wq": ".attn.to_q", ".attn.wk": ".attn.to_k", ".attn.wv": ".attn.to_v",
".attn.gate": ".attn.to_gate", ".attn.wo": ".attn.to_out.0",
}
for k in sdk:
if not (k.startswith("diffusion_model.") and k.endswith(".weight")):
continue
name = k[len("diffusion_model."):-len(".weight")]
if name in krea2_rename:
name = krea2_rename[name]
else:
if name.startswith("blocks."):
name = "transformer_" + name # only the top-level blocks; not txtfusion's *_blocks
for a, b in krea2_sub.items():
name = name.replace(a, b)
names = [name, name[:-2]] if name.endswith(".attn.to_out.0") else [name] # some tools drop the ".0"
for n in names:
key_map[n] = k # bare diffusers name
key_map["transformer.{}".format(n)] = k # diffusers "transformer." prefix
key_map["lycoris_{}".format(n.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: