From 651c0b39b1711f5104af4ea94158966eff37644a Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Thu, 11 Apr 2024 14:51:32 -0400 Subject: [PATCH] clear cache on all exceptions --- .../model_manager/load/model_cache/model_cache_default.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/invokeai/backend/model_manager/load/model_cache/model_cache_default.py b/invokeai/backend/model_manager/load/model_cache/model_cache_default.py index 14e7482050..2ba52d466c 100644 --- a/invokeai/backend/model_manager/load/model_cache/model_cache_default.py +++ b/invokeai/backend/model_manager/load/model_cache/model_cache_default.py @@ -273,7 +273,7 @@ class ModelCache(ModelCacheBase[AnyModel]): snapshot_before = self._capture_memory_snapshot() try: cache_entry.model.to(target_device) - except RuntimeError as e: # blow away cache entry + except Exception as e: # blow away cache entry self._delete_cache_entry(cache_entry) raise e