Skip to content

Commit 6a6d456

Browse files
authored
block info (Comfy-Org#10842)
1 parent 3d1fdaf commit 6a6d456

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

comfy/ldm/qwen_image/model.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -439,7 +439,10 @@ def _forward(
439439
patches = transformer_options.get("patches", {})
440440
blocks_replace = patches_replace.get("dit", {})
441441

442+
transformer_options["total_blocks"] = len(self.transformer_blocks)
443+
transformer_options["block_type"] = "double"
442444
for i, block in enumerate(self.transformer_blocks):
445+
transformer_options["block_index"] = i
443446
if ("double_block", i) in blocks_replace:
444447
def block_wrap(args):
445448
out = {}

0 commit comments

Comments
 (0)