Nodes/ComfyUI-LTX-Attention-Toolkit/LTX Attn — Store Dump
ComfyUI Node

LTX Attn — Store Dump

Make your captures outlive the session

By g-raw·Created 3 months ago·Updated 2 months ago· 2
LTX Attn — Store Dump
    • saved_path
    output_pathltx_store.pt
    include_full_mapstrue
    handle

    Everything the profiling nodes collect lives in memory - in a registry that belongs to the current ComfyUI session. Restart, and it's gone. Store Dump is the node that makes a capture permanent: it saves the AttentionStore and/or QKVStore to a single .pt file on disk, so you can keep a "dev run" and a "distilled run" as reference files and load them back next session. If you're comparing runs made days apart, this is the node that makes it possible.

    It's the pack's persistence layer, and the format is deliberately simple: one file, one dict, with attn and qkv sections (either can be absent). The single handle string resolves both store types - the same handle that Setup Capture output, since attention and QKV data live in independent namespaces and share the name.

    How it works

    • output_path - where the .pt lands. Default ltx_store.pt (relative to ComfyUI's working directory - use an absolute path if you want it somewhere predictable).
    • include_full_maps - default true. Set false to strip the heavy full attention maps and save only metrics and key/query maps, which is the right call for a small archive file or a quick comparison.
    • handle - blank falls back to whichever stores are currently active. An explicit handle that has no attn store under it raises (typo protection - a misspelled name isn't silently swallowed), while a missing QKV store under the same name is silently skipped, because capture_qkv may simply have been off for that run. The dump only errors if neither store resolves.

    Output is a single saved_path STRING so you know exactly where it went.

    Install

    cd ComfyUI/custom_nodes
    git clone https://github.com/g-raw/ComfyUI-LTX-Attention-Toolkit.git
    

    Restart ComfyUI. No extra dependencies, no model files. The pack is work-in-progress, which matters more for this node than most: the README explicitly warns that saved .pt files may stop working after an update. Store formats and node I/O can change with breaking changes, so if you archive captures long-term, pin the repo version and note which version wrote each file.

    Common issues

    The two classic failures: (1) dumping before anything was captured - you get the "no attn or QKV store found under handle" error, which is the pack telling you to run a capture generation first; (2) typoing the handle, which raises loudly (by design, so a wrong name can't silently write an empty file). Also remember the two-run workflow: the store is only populated once the KSampler finishes, so dump in a later queue pass, not the same run as the capture.

    Categoryg_raw/LTX/Profiler

    Inputs (3)

    NameTypeDefaultDescription
    output_pathSTRINGltx_store.pt
    include_full_mapsBOOLEANtrue
    handleSTRING

    Outputs (1)

    NameTypeDescription
    saved_pathSTRING