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

use register instead of heuristic import, get rid of typing warnings

Browse Source
This commit is contained in:
Brandon Rising 2024-03-07 17:07:40 -05:00 committed by Mary Hipp Rogers
parent 9063b1ae61
commit 42d606f07c
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
invokeai/app/services/model_install/model_install_default.py
View File

@ -307,12 +307,19 @@ class ModelInstallService(ModelInstallServiceBase):
), f"This script works on version 3.0.0 yaml files, but your configuration points to a {stanza['version']} version" ), f"This script works on version 3.0.0 yaml files, but your configuration points to a {stanza['version']} version"
continue continue
base_type, model_type, model_name = str(model_key).split("/") _, _, model_name = str(model_key).split("/")
model_path = stanza["path"] import pathlib
model_path = pathlib.Path(stanza["path"])
if not model_path.is_absolute():
model_path = (self._app_config.models_path / model_path)
model_path.resolve()
description = stanza["description"] description = stanza["description"]
model_info = {"name": model_name, "description":description } model_info = {"name": model_name, "description":description }
self.heuristic_import(source=model_path, config=model_info, inplace=True) try:
self.register_path(model_path=model_path, config=model_info)
except Exception as e:
self._logger.warning(f"Model at {model_path} could not be loaded into database: {e}")
def scan_directory(self, scan_dir: Path, install: bool = False) -> List[str]: # noqa D102 def scan_directory(self, scan_dir: Path, install: bool = False) -> List[str]: # noqa D102