Nodes/ComfyUI-LTX-Attention-Toolkit/LTX Attn — Setup Capture
ComfyUI Node

LTX Attn — Setup Capture

A ComfyUI node in g_raw/LTX/Profiler with 14 inputs and 2 outputs.

By g-raw·Created 2 months ago·Updated 29 days ago· 2
LTX Attn — Setup Capture
  • model
  • patched_model
  • handle
capture_satrue
capture_catrue
target_blocks0,8,16,24,32,40,47
target_headsall
capture_stepsall
store_modereduced
full_blocks8,16,24,32,40
full_targets
map_downsample1
capture_qkvfalse
qkv_targets
store_name
reset_storetrue
Categoryg_raw/LTX/Profiler

Inputs (14)

NameTypeDefaultDescription
modelMODEL
capture_saBOOLEANtrue
capture_caBOOLEANtrue
target_blocksSTRING0,8,16,24,32,40,47
target_headsSTRINGall
capture_stepsSTRINGall
store_modeCOMBOreducedreduced: metrics + key/query maps only full_fp16: + full attention map for every target block hybrid: full map only for full_blocks, reduced elsewhere
full_blocksSTRING8,16,24,32,40Blocks stored at full resolution (every head) in hybrid mode. A block also listed in full_targets uses full_targets' per-head selection instead (full_targets wins for that one block); full_blocks still applies normally to any block it doesn't cover -- the two can be combined: coarse whole-block capture for exploration (Query Map/Key Map/Zone Analysis) alongside fine per-head capture for Head Freeze targets.
full_targetsSTRINGOptional, hybrid mode only: for the specific blocks listed here, restrict full-map storage to specific (block, head) pairs instead of every head -- saves RAM when you already know which heads you'll feed into Head Freeze. Paste Head Candidates' candidates_csv directly (one 'block,head' per line), or type manually as 'block:head | block:head | ...' (also accepts 'block:h1,h2,...' and 'block:all'). Heads not listed for a block covered here won't have a full map available, so Query Map/Key Map/Zone Analysis on them will error or skip -- use Head Freeze/QKV Transfer for those, or add that block to full_blocks instead (not here) for full multi-head coverage.
map_downsampleINT11–64
capture_qkvBOOLEANfalseAlso capture raw Q/K/V per head into a separate QKV store, for QKV Transfer. Not redundant with store_mode's full attention map: the map only replays the exact historical pattern, raw Q/K/V lets QKV Transfer recombine components from two different generations into a new pattern.
qkv_targetsSTRINGOnly used when capture_qkv is on. Same format as full_targets/Head Freeze's targets: paste Head Candidates' candidates_csv directly (one 'block,head' per line), or type manually as 'block:head | block:h1,h2,... | block:all | ...'. A whole-string 'all' (or 'all:all') captures every block and head. Independent from target_blocks/target_heads above -- raw Q/K/V is much more expensive than the metrics/key/query maps, so this needs its own explicit list.
store_nameSTRING
reset_storeBOOLEANtrue

Outputs (2)

NameTypeDescription
patched_modelMODEL
handleSTRING