Merge 4c0f5f71e8 into 70d7242e57

comfy/cli_args.py

@@ -188,6 +188,8 @@ parser.add_argument("--user-directory", type=is_valid_directory, default=None, h
 
 parser.add_argument("--enable-compress-response-body", action="store_true", help="Enable compressing response body.")
 
+parser.add_argument('--custom-nodes', type=str, nargs='*', help='Specify which custom nodes to load')
+
 if comfy.options.args_parsing:
     args = parser.parse_args()
 else:

main.py

@@ -261,7 +261,14 @@ def start_comfyui(asyncio_loop=None):
     prompt_server = server.PromptServer(asyncio_loop)
     q = execution.PromptQueue(prompt_server)
 
-    nodes.init_extra_nodes(init_custom_nodes=not args.disable_all_custom_nodes)
+    # Parse requested custom nodes from args
+    requested_nodes = None
+    if args.custom_nodes:
+        requested_nodes = [node.strip() for nodes in args.custom_nodes for node in nodes.split(',')]
+        logging.info(f"Loading only requested custom nodes: {requested_nodes}")
+
+    nodes.init_extra_nodes(init_custom_nodes=not args.disable_all_custom_nodes, 
+                          requested_nodes=requested_nodes)
 
     cuda_malloc_warning()
 

nodes.py

@@ -2165,7 +2165,7 @@ def load_custom_node(module_path: str, ignore=set(), module_parent="custom_nodes
         logging.warning(f"Cannot import {module_path} module for custom nodes: {e}")
         return False
 
-def init_external_custom_nodes():
+def init_external_custom_nodes(requested_nodes=None):
     """
     Initializes the external custom nodes.
 
@@ -2174,6 +2174,9 @@ def init_external_custom_nodes():
 
     Returns:
         None
+    
+    Args:
+        requested_nodes (list, optional): List of custom node names to load. If None, loads all nodes.
     """
     base_node_names = set(NODE_CLASS_MAPPINGS.keys())
     node_paths = folder_paths.get_folder_paths("custom_nodes")
@@ -2187,6 +2190,20 @@ def init_external_custom_nodes():
             module_path = os.path.join(custom_node_path, possible_module)
             if os.path.isfile(module_path) and os.path.splitext(module_path)[1] != ".py": continue
             if module_path.endswith(".disabled"): continue
+                
+            # Skip if not in requested nodes
+            module_name = get_module_name(module_path)
+            if requested_nodes is not None:
+                # Check if this module should be loaded based on requested nodes
+                should_load = False
+                for requested_node in requested_nodes:
+                    if requested_node.lower() in module_name.lower():
+                        should_load = True
+                        break
+                if not should_load:
+                    logging.debug(f"Skipping {module_name} as it's not in requested nodes")
+                    continue
+                    
             time_before = time.perf_counter()
             success = load_custom_node(module_path, base_node_names, module_parent="custom_nodes")
             node_import_times.append((time.perf_counter() - time_before, module_path, success))
@@ -2280,11 +2297,11 @@ def init_builtin_extra_nodes():
     return import_failed
 
 
-def init_extra_nodes(init_custom_nodes=True):
+def init_extra_nodes(init_custom_nodes=True, requested_nodes=None):
     import_failed = init_builtin_extra_nodes()
 
     if init_custom_nodes:
-        init_external_custom_nodes()
+        init_external_custom_nodes(requested_nodes)
     else:
         logging.info("Skipping loading of custom nodes")