Skip to content

Commit 003c2c2

Browse files
psychedeliciousMillu
authored andcommitted
feat(nodes): invocation cache reports disabled if max size is 0
1 parent 661b305 commit 003c2c2

File tree

3 files changed

+9
-5
lines changed

3 files changed

+9
-5
lines changed

invokeai/app/services/invocation_cache/invocation_cache_memory.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,16 +75,20 @@ def create_key(self, invocation: BaseInvocation) -> int:
7575
return hash(invocation.json(exclude={"id"}))
7676

7777
def disable(self) -> None:
78+
if self.__max_cache_size == 0:
79+
return
7880
self.__disabled = True
7981

8082
def enable(self) -> None:
83+
if self.__max_cache_size == 0:
84+
return
8185
self.__disabled = False
8286

8387
def get_status(self) -> InvocationCacheStatus:
8488
return InvocationCacheStatus(
8589
hits=self.__hits,
8690
misses=self.__misses,
87-
enabled=not self.__disabled,
91+
enabled=not self.__disabled and self.__max_cache_size > 0,
8892
size=len(self.__cache),
8993
max_size=self.__max_cache_size,
9094
)

invokeai/frontend/web/src/features/queue/hooks/useDisableInvocationCache.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@ export const useDisableInvocationCache = () => {
1717
});
1818

1919
const isDisabled = useMemo(
20-
() => !cacheStatus?.enabled || !isConnected,
21-
[cacheStatus?.enabled, isConnected]
20+
() => !cacheStatus?.enabled || !isConnected || cacheStatus?.max_size === 0,
21+
[cacheStatus?.enabled, cacheStatus?.max_size, isConnected]
2222
);
2323

2424
const disableInvocationCache = useCallback(async () => {

invokeai/frontend/web/src/features/queue/hooks/useEnableInvocationCache.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@ export const useEnableInvocationCache = () => {
1717
});
1818

1919
const isDisabled = useMemo(
20-
() => cacheStatus?.enabled || !isConnected,
21-
[cacheStatus?.enabled, isConnected]
20+
() => cacheStatus?.enabled || !isConnected || cacheStatus?.max_size === 0,
21+
[cacheStatus?.enabled, cacheStatus?.max_size, isConnected]
2222
);
2323

2424
const enableInvocationCache = useCallback(async () => {

0 commit comments

Comments
 (0)