Merge 6518bc2c0a into edee33f55e

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: