- Added `ToonComposerSequentialWrapper` node to handle ToonComposer SLRA matrices for 14B models on 24GB GPUs. - Implemented `4n + 1` chunk calculation logic for ComfyUI-Sequential-Batcher integration. - Forced `torch.float8_e4m3fn` casting on CUDA devices to prevent VRAM OOM during sparse sketch injection. - Maintained strict compatibility with KJNodes Set/Get routing structures. - Created `AGENTS.md` context file for future AI development.
12 lines
331 B
Python
12 lines
331 B
Python
from .tooncomposer_nodes import ToonComposerSequentialWrapper
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ToonComposerSequentialWrapper": ToonComposerSequentialWrapper
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ToonComposerSequentialWrapper": "ToonComposer SLRA Sequential Wrapper"
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|