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

move incorrectly placed models into correct directory at startup time

Browse Source
This commit is contained in:
Lincoln Stein
2023-09-19 01:18:03 -04:00
parent 73bc088fa7
commit de666fd7bc
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/install/check_root.py
View File

@ -8,7 +8,7 @@ from invokeai.app.services.config import InvokeAIAppConfig
def check_invokeai_root(config: InvokeAIAppConfig):
try:
assert config.model_conf_path.exists(), f"{config.model_conf_path} not found"
assert config.model_conf_path.parent.exists(), f"{config.model_conf_path.parent} not found"
assert config.db_path.parent.exists(), f"{config.db_path.parent} not found"
assert config.models_path.exists(), f"{config.models_path} not found"
if not config.ignore_missing_core_models:

6
invokeai/backend/model_manager/install.py
View File

@ -529,6 +529,9 @@ class ModelInstall(ModelInstallBase):
return old_path
new_path = models_dir / model.base_model.value / model.model_type.value / model.name
self._logger.info(
f"{old_path.name} is not in the right directory for a model of its type. Moving to {new_path}."
)
model.path = self._move_model(old_path, new_path).as_posix()
self._store.update_model(key, model)
return model
@ -652,7 +655,8 @@ class ModelInstall(ModelInstallBase):
def _scan_register(self, model: Path) -> bool:
try:
id = self.register_path(model)
self._logger.info(f"Registered {model} with id {id}")
self.sync_model_path(id) # possibly move it to right place in `models`
self._logger.info(f"Registered {model.name} with id {id}")
self._installed.add(id)
except DuplicateModelException:
pass