wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-08 12:32:32 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

fix signature inconsistency

Browse Source
This commit is contained in:
Deluxe233 2026-01-30 22:25:02 -05:00
parent c6b6128b2b
commit 5bad474118
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy_execution/caching.py
View File

@ -221,7 +221,7 @@ class CacheKeySetInputSignature(CacheKeySet):
logging.warning(f"Node {node_id} cannot be cached due to whatever this thing is: {inputs[key]}") logging.warning(f"Node {node_id} cannot be cached due to whatever this thing is: {inputs[key]}")
node_inputs[key] = Unhashable() node_inputs[key] = Unhashable()
else: else:
node_inputs[key] = inputs[key] node_inputs[key] = [inputs[key]]
self.ancestry_cache[node_id] = ancestors self.ancestry_cache[node_id] = ancestors
return self.ancestry_cache[node_id], node_inputs return self.ancestry_cache[node_id], node_inputs
@ -427,7 +427,7 @@ class LRUCache(BasicCache):
self.ancestry_cache.clear() self.ancestry_cache.clear()
while len(self.cache) > self.max_size and self.min_generation < self.generation: while len(self.cache) > self.max_size and self.min_generation < self.generation:
self.min_generation += 1 self.min_generation += 1
to_remove = [key for key in self.cache if self.used_generation[key] < self.min_generation] to_remove = [key for key in self.cache if key not in self.used_generation or self.used_generation[key] < self.min_generation]
for key in to_remove: for key in to_remove:
del self.cache[key] del self.cache[key]
del self.used_generation[key] del self.used_generation[key]
@ -462,8 +462,8 @@ class LRUCache(BasicCache):
return self return self
async def update_cache_key(self, node_id): async def update_cache_key(self, node_id):
self._mark_used(node_id)
await self._update_cache_key_immediate(node_id) await self._update_cache_key_immediate(node_id)
self._mark_used(node_id)
def is_key_updated(self, node_id): def is_key_updated(self, node_id):
self._mark_used(node_id) self._mark_used(node_id)