wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-25 22:00:19 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Minor refactor of IsChangedCache.get

Browse Source
This commit is contained in:
Jacob Segal 2024-04-21 21:39:52 -07:00
parent b3e547f22b
commit fa48ad3a1f
1 changed files with 22 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
execution.py
View File

@ -33,24 +33,28 @@ class IsChangedCache:
self.is_changed = {} self.is_changed = {}
def get(self, node_id): def get(self, node_id):
if node_id not in self.is_changed: if node_id in self.is_changed:
return self.is_changed[node_id]
node = self.dynprompt.get_node(node_id) node = self.dynprompt.get_node(node_id)
class_type = node["class_type"] class_type = node["class_type"]
class_def = nodes.NODE_CLASS_MAPPINGS[class_type] class_def = nodes.NODE_CLASS_MAPPINGS[class_type]
if hasattr(class_def, "IS_CHANGED"): if not hasattr(class_def, "IS_CHANGED"):
self.is_changed[node_id] = False
return self.is_changed[node_id]
if "is_changed" in node: if "is_changed" in node:
self.is_changed[node_id] = node["is_changed"] self.is_changed[node_id] = node["is_changed"]
else: return self.is_changed[node_id]
input_data_all, _ = get_input_data(node["inputs"], class_def, node_id, self.outputs_cache) input_data_all, _ = get_input_data(node["inputs"], class_def, node_id, self.outputs_cache)
try: try:
is_changed = map_node_over_list(class_def, input_data_all, "IS_CHANGED") is_changed = map_node_over_list(class_def, input_data_all, "IS_CHANGED")
node["is_changed"] = [None if isinstance(x, ExecutionBlocker) else x for x in is_changed] node["is_changed"] = [None if isinstance(x, ExecutionBlocker) else x for x in is_changed]
self.is_changed[node_id] = node["is_changed"]
except: except:
node["is_changed"] = float("NaN") node["is_changed"] = float("NaN")
finally:
self.is_changed[node_id] = node["is_changed"] self.is_changed[node_id] = node["is_changed"]
else:
self.is_changed[node_id] = False
return self.is_changed[node_id] return self.is_changed[node_id]
class CacheSet: class CacheSet: