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 model probing for controlnet checkpoint legacy config files

Browse Source
This commit is contained in:
Lincoln Stein
2023-11-25 15:53:22 -05:00
parent 19baea1883
commit ec510d34b5
5 changed files with 269 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -127,7 +127,7 @@ class ModelInstallService(ModelInstallServiceBase):
model_path = Path(model_path)
metadata = metadata or {}
if metadata.get('source') is None:
metadata['source'] = model_path.as_posix()
metadata['source'] = model_path.resolve().as_posix()
return self._register(model_path, metadata)
def install_path(
@ -138,7 +138,7 @@ class ModelInstallService(ModelInstallServiceBase):
model_path = Path(model_path)
metadata = metadata or {}
if metadata.get('source') is None:
metadata['source'] = model_path.as_posix()
metadata['source'] = model_path.resolve().as_posix()
info: AnyModelConfig = self._probe_model(Path(model_path), metadata)
@ -366,6 +366,7 @@ class ModelInstallService(ModelInstallServiceBase):
# add 'main' specific fields
if hasattr(info, 'config'):
# make config relative to our root
info.config = self.app_config.legacy_conf_dir / info.config
legacy_conf = (self.app_config.root_dir / self.app_config.legacy_conf_dir / info.config).resolve()
info.config = legacy_conf.relative_to(self.app_config.root_dir).as_posix()
self.record_store.add_model(key, info)
return key