From 14c8738a71efc9ac26f89978c5f807861467b4bb Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Thu, 9 Mar 2023 21:41:45 -0500 Subject: [PATCH] fix dangling reference to _model_to_cpu and missing variable model_description --- invokeai/backend/model_management/model_manager.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/invokeai/backend/model_management/model_manager.py b/invokeai/backend/model_management/model_manager.py index 7639e79362..4627f283f5 100644 --- a/invokeai/backend/model_management/model_manager.py +++ b/invokeai/backend/model_management/model_manager.py @@ -500,7 +500,6 @@ class ModelManager(object): print(f">> Offloading {model_name} to CPU") model = self.models[model_name]["model"] model.offload_all() - self.models[model_name]["model"] = self._model_to_cpu(model) gc.collect() if self._has_cuda(): @@ -558,7 +557,7 @@ class ModelManager(object): """ model_name = model_name or Path(repo_or_path).stem model_description = ( - model_description or f"Imported diffusers model {model_name}" + description or f"Imported diffusers model {model_name}" ) new_config = dict( description=model_description,