From 7dfa135b2c5b1b293c53caead6ed6538968d991d Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Wed, 10 May 2023 15:29:10 +1000 Subject: [PATCH] fix(nodes): fix #3306 Check if the cache has the object before deleting it. --- invokeai/app/services/image_storage.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/invokeai/app/services/image_storage.py b/invokeai/app/services/image_storage.py index c5e5cf8a8e..e2593dd473 100644 --- a/invokeai/app/services/image_storage.py +++ b/invokeai/app/services/image_storage.py @@ -270,4 +270,5 @@ class DiskImageStorage(ImageStorageBase): ) # TODO: this should refresh position for LRU cache if len(self.__cache) > self.__max_cache_size: cache_id = self.__cache_ids.get() - del self.__cache[cache_id] + if cache_id in self.__cache: + del self.__cache[cache_id]