From b2e08bb9633044527281609db37b35c89e6926ef Mon Sep 17 00:00:00 2001
From: azazeal04 <132445160+azazeal04@users.noreply.github.com>
Date: Sat, 4 Apr 2026 20:05:17 +0200
Subject: [PATCH] Avoid overwriting existing keys in state_dict

Prevent overwriting existing keys when mapping t_embedder_2 keys to t_embedder.
---
 comfy/utils.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/comfy/utils.py b/comfy/utils.py
index 0a5aaf3d0..822225382 100644
--- a/comfy/utils.py
+++ b/comfy/utils.py
@@ -825,8 +825,9 @@ def twinflow_z_image_key_mapping(state_dict, key):
     """
     if key.startswith("t_embedder_2."):
         new_key = key.replace("t_embedder_2.", "t_embedder.", 1)
-        state_dict[new_key] = state_dict.pop(key)
-    return state_dict
+        if new_key not in state_dict:
+            state_dict[new_key] = state_dict.pop(key)
+    return state_dic
 
 def repeat_to_batch_size(tensor, batch_size, dim=0):
     if tensor.shape[dim] > batch_size: