From 33f83d53ae3897962b8248c12276759192fde0c0 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Tue, 24 Feb 2026 13:02:05 -0800 Subject: [PATCH] Fix KeyError when prompt entries lack class_type key (#12595) Skip entries in the prompt dict that don't contain a class_type key in apply_replacements(), preventing crashes on metadata or non-node entries. Fixes Comfy-Org/ComfyUI#12517 --- app/node_replace_manager.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/app/node_replace_manager.py b/app/node_replace_manager.py index 03b603c70..d9aab5b22 100644 --- a/app/node_replace_manager.py +++ b/app/node_replace_manager.py @@ -46,6 +46,8 @@ class NodeReplaceManager: connections: dict[str, list[tuple[str, str, int]]] = {} need_replacement: set[str] = set() for node_number, node_struct in prompt.items(): + if "class_type" not in node_struct or "inputs" not in node_struct: + continue class_type = node_struct["class_type"] # need replacement if not in NODE_CLASS_MAPPINGS and has replacement if class_type not in nodes.NODE_CLASS_MAPPINGS.keys() and self.has_replacement(class_type):