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

reinitialize models.yaml if corrupt or missing

Browse Source
This commit is contained in:
Lincoln Stein 2023-07-20 11:26:20 -04:00
parent a2e2f0858d
commit b1a6ba552b
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/backend/install/invokeai_configure.py
View File

@ -23,6 +23,7 @@ from urllib import request
import npyscreen import npyscreen
import transformers import transformers
import omegaconf
from diffusers import AutoencoderKL from diffusers import AutoencoderKL
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
from huggingface_hub import HfFolder from huggingface_hub import HfFolder
@ -568,7 +569,14 @@ def default_startup_options(init_file: Path) -> Namespace:
return opts return opts
def default_user_selections(program_opts: Namespace) -> InstallSelections: def default_user_selections(program_opts: Namespace) -> InstallSelections:
try:
installer = ModelInstall(config) installer = ModelInstall(config)
except omegaconf.errors.ConfigKeyError:
logger.warning('Your models.yaml file is corrupt or out of date. Reinitializing')
initialize_rootdir(config.root_path, True)
installer = ModelInstall(config)
models = installer.all_models() models = installer.all_models()
return InstallSelections( return InstallSelections(
install_models=[models[installer.default_model()].path or models[installer.default_model()].repo_id] install_models=[models[installer.default_model()].path or models[installer.default_model()].repo_id]

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

@ -958,7 +958,6 @@ class ModelManager(object):
config.controlnet_dir, config.controlnet_dir,
] if x ] if x
} }
print(directories)
scanner = ScanAndImport(directories, self.logger, ignore=known_paths, installer=installer) scanner = ScanAndImport(directories, self.logger, ignore=known_paths, installer=installer)
scanner.search() scanner.search()