Skip to content

Commit 1768392

Browse files
committed
Use is_link for checking if an input is a link instead of looking for list status
1 parent 962f62d commit 1768392

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

app/node_replace_manager.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
if TYPE_CHECKING:
77
from comfy_api.latest._node_replace import NodeReplace
88

9+
from comfy_execution.graph_utils import is_link
910
from nodes import NODE_CLASS_MAPPINGS
1011

1112
class NodeStruct(TypedDict):
@@ -51,7 +52,7 @@ def apply_replacements(self, prompt: dict[str, NodeStruct]):
5152
need_replacement.add(node_number)
5253
# keep track of connections
5354
for input_id, input_value in node_struct["inputs"].items():
54-
if isinstance(input_value, list):
55+
if is_link(input_value):
5556
conn_number = input_value[0]
5657
connections.setdefault(conn_number, []).append((node_number, input_id, input_value[1]))
5758
if len(need_replacement) > 0:

0 commit comments

Comments
 (0)