From 472309ff2d1cc50d23191804cbd21c0b6d7c783e Mon Sep 17 00:00:00 2001 From: Haoming Date: Mon, 18 May 2026 23:00:47 +0800 Subject: [PATCH] keys --- extensions-builtin/sd_forge_lora/networks.py | 2 ++ modules_forge/packages/comfy/lora.py | 9 +++++++-- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/extensions-builtin/sd_forge_lora/networks.py b/extensions-builtin/sd_forge_lora/networks.py index 0bf04ad8..ae55724d 100644 --- a/extensions-builtin/sd_forge_lora/networks.py +++ b/extensions-builtin/sd_forge_lora/networks.py @@ -31,6 +31,8 @@ def process_anima(lora: dict[str, torch.Tensor]): for k in keys: if k.startswith("diffusion_model.llm_adapter"): lora[k.replace("diffusion_model", "text_encoders.qwen3_06b")] = lora.pop(k) + elif k.startswith("lora_unet_llm_adapter"): + lora[k.replace("lora_unet_llm_adapter", "lora_te_llm_adapter")] = lora.pop(k) def load_lora_for_models(model: "UnetPatcher", clip: "CLIP", lora: dict[str, torch.Tensor], strength_model: float, strength_clip: float, filename: str = "default", online_mode: bool = False): diff --git a/modules_forge/packages/comfy/lora.py b/modules_forge/packages/comfy/lora.py index 741205f4..11c340df 100644 --- a/modules_forge/packages/comfy/lora.py +++ b/modules_forge/packages/comfy/lora.py @@ -165,10 +165,15 @@ def model_lora_keys_clip(model, key_map={}): key_map["lora_te{}_{}".format(t5_index, l_key.replace(".", "_"))] = k - for k in sdk: - if k.endswith(".weight") and k.startswith("qwen3_06b."): # Anima + for k in sdk: # Anima + if not k.endswith(".weight"): + continue + if k.startswith("qwen3_06b.model"): _key = k[len("qwen3_06b.model.layers.") : -len(".weight")] key_map["lora_te_layers_{}".format(_key.replace(".", "_"))] = k + elif k.startswith("qwen3_06b.llm_adapter"): + _key = k[len("qwen3_06b.") : -len(".weight")] + key_map["lora_te_{}".format(_key.replace(".", "_"))] = k return key_map