fix keyerror bug that was causing merge script to crash

Lincoln Stein · hipsterusername · commit f4ead5e07f49 · 2023-07-30T19:25:44.000-04:00
diff --git a/invokeai/frontend/merge/merge_diffusers.py b/invokeai/frontend/merge/merge_diffusers.py
@@ -320,7 +320,7 @@ def validate_field_values(self) -> bool:
 
     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"
         ]

Original file line number	Diff line number	Diff line change
`@@ -320,7 +320,7 @@ def validate_field_values(self) -> bool:`
`320`	`320`
`321`	`321`	`def get_model_names(self, base_model: BaseModelType = None) -> List[str]:`
`322`	`322`	`model_names = [`
`323`		`- info["name"]`
	`323`	`+ info["model_name"]`
`324`	`324`	`for info in self.model_manager.list_models(model_type=ModelType.Main, base_model=base_model)`
`325`	`325`	`if info["model_format"] == "diffusers"`
`326`	`326`	`]`