From 84c983a779ae80dc9b56d6a8c47a441d75124e73 Mon Sep 17 00:00:00 2001
From: azazeal04 <132445160+azazeal04@users.noreply.github.com>
Date: Sat, 4 Apr 2026 15:57:19 +0200
Subject: [PATCH] Add detection for TwinFlow-Z-Image checkpoints

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

diff --git a/comfy/model_detection.py b/comfy/model_detection.py
index 1c8ae2325..aed486e09 100644
--- a/comfy/model_detection.py
+++ b/comfy/model_detection.py
@@ -44,6 +44,12 @@ def calculate_transformer_depth(prefix, state_dict_keys, state_dict):
 def detect_unet_config(state_dict, key_prefix, metadata=None):
     state_dict_keys = list(state_dict.keys())
 
+    # TwinFlow-Z-Image: detect dual timestep embedder checkpoints first.
+    if any(k.startswith('{}t_embedder_2.'.format(key_prefix)) for k in state_dict_keys):
+        return {
+            "image_model": "twinflow_z_image",
+            "architecture": "TwinFlow_Z_Image",
+        }
     if '{}joint_blocks.0.context_block.attn.qkv.weight'.format(key_prefix) in state_dict_keys: #mmdit model
         unet_config = {}
         unet_config["in_channels"] = state_dict['{}x_embedder.proj.weight'.format(key_prefix)].shape[1]