Also skip prompt entries missing inputs key

Amp-Thread-ID: https://ampcode.com/threads/T-019c8ae9-0bc0-77de-8bde-a4a7de38cff2
Co-authored-by: Amp <amp@ampcode.com>
This commit is contained in:
Jedrzej Kosinski 2026-02-23 06:38:39 -08:00
parent 13dc544db2
commit 0863d8acb1

View File

@ -46,7 +46,7 @@ class NodeReplaceManager:
connections: dict[str, list[tuple[str, str, int]]] = {} connections: dict[str, list[tuple[str, str, int]]] = {}
need_replacement: set[str] = set() need_replacement: set[str] = set()
for node_number, node_struct in prompt.items(): for node_number, node_struct in prompt.items():
if "class_type" not in node_struct: if "class_type" not in node_struct or "inputs" not in node_struct:
continue continue
class_type = node_struct["class_type"] class_type = node_struct["class_type"]
# need replacement if not in NODE_CLASS_MAPPINGS and has replacement # need replacement if not in NODE_CLASS_MAPPINGS and has replacement