| # SPDX-License-Identifier: MIT | |
| """ | |
| 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 | |
| """ | |
| try: | |
| # Normal case: loaded as a package from ComfyUI/custom_nodes. | |
| from .turboquant_nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS | |
| except ImportError: | |
| # Loaded as a top-level module (pytest imports this file as "__init__" | |
| # because the repository root is a package, and flat loaders do the same). | |
| from turboquant_nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS | |
| __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] | |
| WEB_DIRECTORY = None | |