Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(nodes): do not disable invocation cache delete methods

Browse Source 
When the runtime disabled flag is on, do not skip the delete methods. This could lead to a hit on a missing resource.

Do skip them when the cache size is 0, because the user cannot change this (must restart app to change it).
This commit is contained in:
psychedelicious 2023-09-22 16:48:58 +10:00 committed by Kent Keirsey
parent 60b3c6a201
commit 7d683b4db6
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/app/services/invocation_cache/invocation_cache_memory.py
View File

@ -56,14 +56,14 @@ class MemoryInvocationCache(InvocationCacheBase):
pass pass
def delete(self, key: Union[int, str]) -> None: def delete(self, key: Union[int, str]) -> None:
if self.__max_cache_size == 0 or self.__disabled: if self.__max_cache_size == 0:
return return
if key in self.__cache: if key in self.__cache:
del self.__cache[key] del self.__cache[key]
def clear(self, *args, **kwargs) -> None: def clear(self, *args, **kwargs) -> None:
if self.__max_cache_size == 0 or self.__disabled: if self.__max_cache_size == 0:
return return
self.__cache.clear() self.__cache.clear()
@ -94,7 +94,7 @@ class MemoryInvocationCache(InvocationCacheBase):
) )
def _delete_by_match(self, to_match: str) -> None: def _delete_by_match(self, to_match: str) -> None:
if self.__max_cache_size == 0 or self.__disabled: if self.__max_cache_size == 0:
return return
keys_to_delete = set() keys_to_delete = set()