We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3d1fdaf commit 6a6d456Copy full SHA for 6a6d456
comfy/ldm/qwen_image/model.py
@@ -439,7 +439,10 @@ def _forward(
439
patches = transformer_options.get("patches", {})
440
blocks_replace = patches_replace.get("dit", {})
441
442
+ transformer_options["total_blocks"] = len(self.transformer_blocks)
443
+ transformer_options["block_type"] = "double"
444
for i, block in enumerate(self.transformer_blocks):
445
+ transformer_options["block_index"] = i
446
if ("double_block", i) in blocks_replace:
447
def block_wrap(args):
448
out = {}
0 commit comments