From 00c4e72a9715ea0fefa5c5850c9ffa8f44f646ef Mon Sep 17 00:00:00 2001
From: azazeal04 <132445160+azazeal04@users.noreply.github.com>
Date: Sat, 4 Apr 2026 15:49:32 +0200
Subject: [PATCH] Implement TwinFlow-Z-Image LoRA compatibility

Add compatibility for TwinFlow-Z-Image LoRAs to target t_embedder_2.* keys.
---
 comfy/lora.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/comfy/lora.py b/comfy/lora.py
index 63ee85323..f6f1fb150 100644
--- a/comfy/lora.py
+++ b/comfy/lora.py
@@ -328,6 +328,13 @@ def model_lora_keys_unet(model, key_map={}):
                 key_map["lycoris_{}".format(key_lora.replace(".", "_"))] = to
                 key_map[key_lora] = to
 
+    # TwinFlow-Z-Image LoRAs can target t_embedder_2.* keys.
+    # Alias them back to t_embedder.* targets for compatibility.
+    if isinstance(model, comfy.model_base.TwinFlow_Z_Image):
+        for key in list(key_map.keys()):
+            if "t_embedder." in key and "t_embedder_2." not in key:
+                key_map[key.replace("t_embedder.", "t_embedder_2.", 1)] = key_map[key]
+                
     if isinstance(model, comfy.model_base.Kandinsky5):
         for k in sdk:
             if k.startswith("diffusion_model.") and k.endswith(".weight"):