Extend ComfyUI type definitions in __init__.py

Added support for extending ComfyUI type definitions with custom node type registration for TILE_CONFIG.
2026-01-17 19:56:03 +08:00
parent a6110f9969
commit efedb19f45
1 changed files with 8 additions and 0 deletions
@@ -11,4 +11,12 @@ NODE_DISPLAY_NAME_MAPPINGS = {
    "CustomTileMerger": "🔧 Advanced Tile Merger",
 }

+# Extend ComfyUI Type Definitions
+try:
+    from comfy.graph_utils import GraphBuilder
+    # Register custom dict type check
+    GraphBuilder.add_node_type("TILE_CONFIG", lambda x: isinstance(x, dict) and "splits" in x)
+except ImportError:
+    pass
+
 __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]