From e4e20d79b22eae61eba9cbf9ebc6349294d28e11 Mon Sep 17 00:00:00 2001 From: Jacob Segal Date: Wed, 14 Feb 2024 21:04:50 -0800 Subject: [PATCH] Allow `input_info` to be of type `None` --- comfy/graph.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/graph.py b/comfy/graph.py index 2612317f9..656ebeb97 100644 --- a/comfy/graph.py +++ b/comfy/graph.py @@ -105,7 +105,7 @@ class TopologicalSort: if subgraph_nodes is not None and from_node_id not in subgraph_nodes: continue input_type, input_category, input_info = self.get_input_info(unique_id, input_name) - is_lazy = "lazy" in input_info and input_info["lazy"] + is_lazy = input_info is not None and "lazy" in input_info and input_info["lazy"] if include_lazy or not is_lazy: self.add_strong_link(from_node_id, from_socket, unique_id)