wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-17 00:43:48 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Skip replacements that lead to a missing node_id anyway

Browse Source
This commit is contained in:
Jedrzej Kosinski 2026-02-12 22:29:15 -08:00
parent 5b4ba294a1
commit aac40e4fca
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/node_replace_manager.py
View File

@ -64,9 +64,13 @@ class NodeReplaceManager:
# just use the first replacement # just use the first replacement
replacement = replacements[0] replacement = replacements[0]
new_node_id = replacement.new_node_id new_node_id = replacement.new_node_id
# if replacement is not a valid node, skip trying to replace it as will only cause confusion
if new_node_id not in NODE_CLASS_MAPPINGS.keys():
continue
# first, replace node id (class_type) # first, replace node id (class_type)
new_node_struct = copy_node_struct(node_struct, empty_inputs=True) new_node_struct = copy_node_struct(node_struct, empty_inputs=True)
new_node_struct["class_type"] = new_node_id new_node_struct["class_type"] = new_node_id
# TODO: consider replacing display_name in _meta as well for error reporting purposes; would need to query node schema
# second, replace inputs # second, replace inputs
if replacement.input_mapping is not None: if replacement.input_mapping is not None:
for input_map in replacement.input_mapping: for input_map in replacement.input_mapping: