From 6518bc2c0abf4202a28b06e0ab2835c44daf5daf Mon Sep 17 00:00:00 2001 From: woctordho Date: Fri, 19 Dec 2025 20:07:52 +0800 Subject: [PATCH] Support LoRA from NewBie LoRA trainer --- comfy/lora.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/comfy/lora.py b/comfy/lora.py index 2ed0acb9d..ca6c3cf89 100644 --- a/comfy/lora.py +++ b/comfy/lora.py @@ -322,6 +322,10 @@ def model_lora_keys_unet(model, key_map={}): key_map["diffusion_model.{}".format(key_lora)] = to key_map["transformer.{}".format(key_lora)] = to key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = to + for k in sdk: + if k.startswith("diffusion_model.") and k.endswith(".weight"): + key_lora = k[len("diffusion_model."):-len(".weight")] + key_map["base_model.model.{}".format(key_lora)] = k # NewBie LoRA trainer if isinstance(model, comfy.model_base.Kandinsky5): for k in sdk: