298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revert "Reduce repeated calls of get_immediate_node_signature for ancestors in cache (#4871)"

Browse Source 
This reverts commit f6b7194f64.
This commit is contained in:
comfyanonymous 2024-09-13 23:24:47 -04:00
parent cf80d28689
commit b3ce8fb9fd
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_execution/caching.py
View File

@ -67,7 +67,6 @@ class CacheKeySetInputSignature(CacheKeySet):
super().__init__(dynprompt, node_ids, is_changed_cache) super().__init__(dynprompt, node_ids, is_changed_cache)
self.dynprompt = dynprompt self.dynprompt = dynprompt
self.is_changed_cache = is_changed_cache self.is_changed_cache = is_changed_cache
self.immediate_node_signature = {}
self.add_keys(node_ids) self.add_keys(node_ids)
def include_node_id_in_input(self) -> bool: def include_node_id_in_input(self) -> bool:
@ -95,8 +94,6 @@ class CacheKeySetInputSignature(CacheKeySet):
if not dynprompt.has_node(node_id): if not dynprompt.has_node(node_id):
# This node doesn't exist -- we can't cache it. # This node doesn't exist -- we can't cache it.
return [float("NaN")] return [float("NaN")]
if node_id in self.immediate_node_signature: # reduce repeated calls of ancestors
return self.immediate_node_signature[node_id]
node = dynprompt.get_node(node_id) node = 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]
@ -111,7 +108,6 @@ class CacheKeySetInputSignature(CacheKeySet):
signature.append((key,("ANCESTOR", ancestor_index, ancestor_socket))) signature.append((key,("ANCESTOR", ancestor_index, ancestor_socket)))
else: else:
signature.append((key, inputs[key])) signature.append((key, inputs[key]))
self.immediate_node_signature[node_id] = signature
return signature return signature
# This function returns a list of all ancestors of the given node. The order of the list is # This function returns a list of all ancestors of the given node. The order of the list is