From a72c7edd3c51ec7334cb14789e01890ef6ef596c Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Thu, 27 Nov 2025 01:48:48 +0200
Subject: [PATCH] Support official LoRAs

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

diff --git a/comfy/lora.py b/comfy/lora.py
index 3a9077869..e7202ce97 100644
--- a/comfy/lora.py
+++ b/comfy/lora.py
@@ -322,6 +322,13 @@ def model_lora_keys_unet(model, key_map={}):
                 key_map["diffusion_model.{}".format(key_lora)] = to
                 key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = to
 
+    if isinstance(model, comfy.model_base.Kandinsky5):
+        for k in sdk:
+            if k.startswith("diffusion_model.") and k.endswith(".weight"):
+                key_lora = k[len("diffusion_model."):-len(".weight")]
+                key_map["{}".format(key_lora)] = k
+                key_map["transformer.{}".format(key_lora)] = k
+
     return key_map