From f4ead5e07f49b456bc1a784786819c179c72e991 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Sun, 30 Jul 2023 18:35:43 -0400 Subject: [PATCH] fix keyerror bug that was causing merge script to crash --- invokeai/frontend/merge/merge_diffusers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/invokeai/frontend/merge/merge_diffusers.py b/invokeai/frontend/merge/merge_diffusers.py index 81a7b3f6c8..db493ec17f 100644 --- a/invokeai/frontend/merge/merge_diffusers.py +++ b/invokeai/frontend/merge/merge_diffusers.py @@ -320,7 +320,7 @@ class mergeModelsForm(npyscreen.FormMultiPageAction): def get_model_names(self, base_model: BaseModelType = None) -> List[str]: model_names = [ - info["name"] + info["model_name"] for info in self.model_manager.list_models(model_type=ModelType.Main, base_model=base_model) if info["model_format"] == "diffusers" ]