From b92f876284440add1745ea06158ab5bc60ed0088 Mon Sep 17 00:00:00 2001 From: Jacob Segal Date: Thu, 29 Aug 2024 00:16:30 -0700 Subject: [PATCH] Change entangled nodes to only get actual types --- comfy_execution/graph.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/comfy_execution/graph.py b/comfy_execution/graph.py index 40eaf00dd6e9..93e83718c91e 100644 --- a/comfy_execution/graph.py +++ b/comfy_execution/graph.py @@ -154,7 +154,19 @@ def resolve_dynamic_definitions(self, node_id_set: Set[str]): if hasattr(class_def, "resolve_dynamic_types"): entangled_types = {} for (entangled_id, entangled_name) in entangled.get(node_id, []): - input_types, output_types = self.get_input_output_types(entangled_id) + entangled_def = self.get_node_definition(entangled_id) + if entangled_def is None: + continue + input_types = {} + output_types = {} + for input_category, input_list in entangled_def["input"].items(): + for input_name, input_info in input_list.items(): + if isinstance(input_info, tuple) or input_category != "hidden": + input_types[input_name] = input_info[0] + for i in range(len(entangled_def["output"])): + output_name = entangled_def["output_name"][i] + output_types[output_name] = entangled_def["output"][i] + if entangled_name not in entangled_types: entangled_types[entangled_name] = [] entangled_types[entangled_name].append({