Custom ComfyUI node that compresses attention KV cache by 4.6x using TurboQuant (Google ICLR 2026) 3-bit quantization. Enables LTX-2.3 22B to run on a single V100-SXM2-32GB with 6GB headroom. Verified: 41 frames generated, 26.5GB peak VRAM (down from 32.4GB). First known TQ3 KV compression deployment in a video generation pipeline. Nodes: TurboQuantPatch (model patcher), TurboQuantInfo (stats display) Quality: 0.983 cosine similarity, 4.57x compression ratio Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
17 lines
466 B
Python
17 lines
466 B
Python
"""
|
|
ComfyUI-TurboQuant: TQ3 KV Cache Compression
|
|
|
|
Reduces attention KV cache VRAM by ~4.5x using 3-bit Lloyd-Max
|
|
quantization with Fast Walsh-Hadamard Transform decorrelation.
|
|
|
|
Nodes:
|
|
- TurboQuant KV Patch: Patches any model for compressed KV cache
|
|
- TurboQuant Info: Shows compression stats
|
|
"""
|
|
|
|
from .turboquant_nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|
|
|
|
WEB_DIRECTORY = None
|