From e9c311b2458a327585a7e387558377c8190eebb0 Mon Sep 17 00:00:00 2001
From: Rainer <rainer@rainerots.eu>
Date: Fri, 1 May 2026 02:33:41 +0300
Subject: [PATCH] OneTainer ERNIE LoRA support (#13640)

---
 comfy/lora.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/comfy/lora.py b/comfy/lora.py
index 63ee85323..e4337c729 100644
--- a/comfy/lora.py
+++ b/comfy/lora.py
@@ -342,6 +342,12 @@ def model_lora_keys_unet(model, key_map={}):
                 key_map["base_model.model.{}".format(key_lora)] = k  # Official base model loras
                 key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = k  # LyCORIS/LoKR format
 
+    if isinstance(model, comfy.model_base.ErnieImage):
+        for k in sdk:
+            if k.startswith("diffusion_model.") and k.endswith(".weight"):
+                key_lora = k[len("diffusion_model."):-len(".weight")]
+                key_map["transformer.{}".format(key_lora)] = k
+
     return key_map