AI add middle blocks

This commit is contained in:
Daniel Martinek
2026-04-02 11:04:08 +02:00
parent 81514ee96f
commit 12fdc5e25c
+5
View File
@@ -356,8 +356,13 @@ class SelectiveGuiderMathNode(io.ComfyNode):
h2 = _apply_block_expr(h, transformer_options, stage, idx)
return h2, hsp
def middle_patch(x, extra_options):
stage, idx = _parse_block(extra_options, "middle")
return _apply_block_expr(x, extra_options, stage, idx)
patches["input_block_patch"] = list(patches.get("input_block_patch", [])) + [input_block_patch]
patches["output_block_patch"] = list(patches.get("output_block_patch", [])) + [output_block_patch]
patches["middle_patch"] = list(patches.get("middle_patch", [])) + [middle_patch]
topts["patches"] = patches
patched_dict["transformer_options"] = topts