wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-15 04:52:31 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

feat: add BasicCache.clear_all() public API for cache invalidation

Browse Source
This commit is contained in:
Julián Mulet 2026-04-12 00:23:16 +02:00
parent a134423890
commit cfcd334026
No known key found for this signature in database
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
comfy_execution/caching.py
View File

@ -190,6 +190,17 @@ class BasicCache:
for key in to_remove: for key in to_remove:
del self.subcaches[key] del self.subcaches[key]
def clear_all(self):
"""Drop all cached outputs unconditionally.
This is the public API for external subsystems (e.g. aggressive model
offloading) that need to invalidate every cached result — for instance
after model parameters have been moved to the ``meta`` device and the
cached tensors are no longer usable.
"""
self.cache.clear()
self.subcaches.clear()
def clean_unused(self): def clean_unused(self):
assert self.initialized assert self.initialized
self._clean_cache() self._clean_cache()