Update invokeai/backend/model_management/model_cache.py

lstein · StAlKeR7779 · web-flow · commit bb1b8ceaa8de · 2023-08-16T08:48:44.000-04:00
Co-authored-by: StAlKeR7779 &lt;stalkek7779@yandex.ru&gt;
diff --git a/invokeai/backend/model_management/model_cache.py b/invokeai/backend/model_management/model_cache.py
@@ -226,7 +226,7 @@ def get_model(
             self.stats.high_watermark = max(self.stats.high_watermark, self._cache_size())
             self.stats.in_cache = len(self._cached_models)
             self.stats.loaded_model_sizes[key] = max(
-                self.stats.loaded_model_sizes.get("key", 0), model_info.get_size(submodel)
+                self.stats.loaded_model_sizes.get(key, 0), model_info.get_size(submodel)
             )
 
         with suppress(Exception):

Original file line number	Diff line number	Diff line change
`@@ -226,7 +226,7 @@ def get_model(`
`226`	`226`	`self.stats.high_watermark = max(self.stats.high_watermark, self._cache_size())`
`227`	`227`	`self.stats.in_cache = len(self._cached_models)`
`228`	`228`	`self.stats.loaded_model_sizes[key] = max(`
`229`		`- self.stats.loaded_model_sizes.get("key", 0), model_info.get_size(submodel)`
	`229`	`+ self.stats.loaded_model_sizes.get(key, 0), model_info.get_size(submodel)`
`230`	`230`	`)`
`231`	`231`
`232`	`232`	`with suppress(Exception):`