From 13dc544db27dc82d3c8e49625c346acc70aef477 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Mon, 23 Feb 2026 06:35:39 -0800 Subject: [PATCH] Fix KeyError when prompt entries lack class_type key 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 Amp-Thread-ID: https://ampcode.com/threads/T-019c8ae9-0bc0-77de-8bde-a4a7de38cff2 Co-authored-by: Amp --- 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..39e5d76df 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: + 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):