From b6c3989eee99d2710b57ff1dae35004859b31e6e Mon Sep 17 00:00:00 2001
From: Haoming <hmstudy02@gmail.com>
Date: Sun, 23 Nov 2025 15:03:39 +0800
Subject: [PATCH] block info

---
 comfy/ldm/qwen_image/model.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/comfy/ldm/qwen_image/model.py b/comfy/ldm/qwen_image/model.py
index 427ea19c1..8c75670cd 100644
--- a/comfy/ldm/qwen_image/model.py
+++ b/comfy/ldm/qwen_image/model.py
@@ -439,7 +439,10 @@ class QwenImageTransformer2DModel(nn.Module):
         patches = transformer_options.get("patches", {})
         blocks_replace = patches_replace.get("dit", {})
 
+        transformer_options["total_blocks"] = len(self.transformer_blocks)
+        transformer_options["block_type"] = "double"
         for i, block in enumerate(self.transformer_blocks):
+            transformer_options["block_index"] = i
             if ("double_block", i) in blocks_replace:
                 def block_wrap(args):
                     out = {}