Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix bug in model_manager that prevented import of inpainting models

Browse Source
This commit is contained in:
Lincoln Stein 2023-06-02 22:39:26 -04:00
parent 91918e648b
commit 72d1e4e404
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_management/model_manager.py
View File

@ -865,7 +865,7 @@ class ModelManager(object):
model_config_file = self.globals.legacy_conf_path / "v1-inference.yaml"
elif model_type == SDLegacyType.V1_INPAINT:
self.logger.debug("SD-v1 inpainting model detected")
model_config_file = self.globals.legacy_conf_path / "v1-inpainting-inference.yaml",
model_config_file = self.globals.legacy_conf_path / "v1-inpainting-inference.yaml"
elif model_type == SDLegacyType.V2_v:
self.logger.debug("SD-v2-v model detected")
model_config_file = self.globals.legacy_conf_path / "v2-inference-v.yaml"

6
invokeai/frontend/install/model_install.py
View File

@ -569,11 +569,13 @@ class addModelsForm(npyscreen.FormMultiPage):
self.subprocess_connection = None
self.monitor.entry_widget.buffer(['** Action Complete **'])
self.display()
# this is crazy
# rebuild the form, saving log messages
saved_messages = self.monitor.entry_widget.values
self.parentApp.main_form = self.parentApp.addForm(
"MAIN", addModelsForm, name="Install Stable Diffusion Models"
)
self.parentApp.switchForm('NEW')
self.parentApp.switchForm('MAIN')
self.parentApp.main_form.monitor.entry_widget.values = saved_messages
return
self.monitor.entry_widget.buffer([data])
self.display()