feat(ace-step): add ACE-Step 1.5 lycoris key alias mapping for LoKR #12638 (#12665)

2026-02-28 02:44:04 +00:00 · 2026-02-27 12:19:19 +13:00
parent 3811780e4f
commit b233dbe0bc
1 changed files with 1 additions and 0 deletions
--- a/comfy/lora.py
+++ b/comfy/lora.py
@@ -337,6 +337,7 @@ def model_lora_keys_unet(model, key_map={}):
            if k.startswith("diffusion_model.decoder.") and k.endswith(".weight"):
                key_lora = k[len("diffusion_model.decoder."):-len(".weight")]
                key_map["base_model.model.{}".format(key_lora)] = k  # Official base model loras
+                key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = k  # LyCORIS/LoKR format

    return key_map