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

feat(nodes): invocation cache reports disabled if max size is 0

Browse Source
This commit is contained in:
psychedelicious 2023-09-21 22:12:11 +10:00 committed by Kent Keirsey
parent 7ac99d6bc3
commit fa54974bff
3 changed files with 9 additions and 5 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

@ -75,16 +75,20 @@ class MemoryInvocationCache(InvocationCacheBase):
return hash(invocation.json(exclude={"id"}))
def disable(self) -> None:
if self.__max_cache_size == 0:
return
self.__disabled = True
def enable(self) -> None:
if self.__max_cache_size == 0:
return
self.__disabled = False
def get_status(self) -> InvocationCacheStatus:
return InvocationCacheStatus(
hits=self.__hits,
misses=self.__misses,
enabled=not self.__disabled,
enabled=not self.__disabled and self.__max_cache_size > 0,
size=len(self.__cache),
max_size=self.__max_cache_size,
)

4
invokeai/frontend/web/src/features/queue/hooks/useDisableInvocationCache.ts
View File

@ -17,8 +17,8 @@ export const useDisableInvocationCache = () => {
});
const isDisabled = useMemo(
() => !cacheStatus?.enabled || !isConnected,
[cacheStatus?.enabled, isConnected]
() => !cacheStatus?.enabled || !isConnected || cacheStatus?.max_size === 0,
[cacheStatus?.enabled, cacheStatus?.max_size, isConnected]
);
const disableInvocationCache = useCallback(async () => {

4
invokeai/frontend/web/src/features/queue/hooks/useEnableInvocationCache.ts
View File

@ -17,8 +17,8 @@ export const useEnableInvocationCache = () => {
});
const isDisabled = useMemo(
() => cacheStatus?.enabled || !isConnected,
[cacheStatus?.enabled, isConnected]
() => cacheStatus?.enabled || !isConnected || cacheStatus?.max_size === 0,
[cacheStatus?.enabled, cacheStatus?.max_size, isConnected]
);
const enableInvocationCache = useCallback(async () => {