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

Resolve when instantiating _cached_model_paths

Browse Source
This commit is contained in:
Brandon Rising 2024-03-15 17:07:14 -04:00 committed by psychedelicious
parent a1743647b7
commit ea5bc94b9c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -328,7 +328,7 @@ class ModelInstallService(ModelInstallServiceBase):
yaml_path.rename(yaml_path.with_suffix(".yaml.bak"))
def scan_directory(self, scan_dir: Path, install: bool = False) -> List[str]: # noqa D102
self._cached_model_paths = {Path(x.path).absolute() for x in self.record_store.all_models()}
self._cached_model_paths = {Path(x.path).resolve() for x in self.record_store.all_models()}
callback = self._scan_install if install else self._scan_register
search = ModelSearch(on_model_found=callback)
self._models_installed.clear()
@ -530,7 +530,7 @@ class ModelInstallService(ModelInstallServiceBase):
return model
def _scan_register(self, model: Path) -> bool:
if any(model == m.resolve() for m in self._cached_model_paths):
if model in self._cached_model_paths:
return True
try:
id = self.register_path(model)