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 duplicate model key addition when root directory is a relative path

Browse Source
This commit is contained in:
Lincoln Stein 2023-06-28 17:02:03 -04:00
parent 8a90e51408
commit ace4f6d586
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -675,13 +675,15 @@ class ModelManager(object):
base_model: Optional[BaseModelType] = None, base_model: Optional[BaseModelType] = None,
model_type: Optional[ModelType] = None, model_type: Optional[ModelType] = None,
): ):
loaded_files = set() loaded_files = set()
new_models_found = False new_models_found = False
self.logger.info(f'scanning {self.app_config.models_path} for new models')
with Chdir(self.app_config.root_path): with Chdir(self.app_config.root_path):
for model_key, model_config in list(self.models.items()): for model_key, model_config in list(self.models.items()):
model_name, cur_base_model, cur_model_type = self.parse_key(model_key) model_name, cur_base_model, cur_model_type = self.parse_key(model_key)
model_path = self.app_config.root_path / model_config.path model_path = self.app_config.root_path.absolute() / model_config.path
if not model_path.exists(): if not model_path.exists():
model_class = MODEL_CLASSES[cur_base_model][cur_model_type] model_class = MODEL_CLASSES[cur_base_model][cur_model_type]
if model_class.save_to_config: if model_class.save_to_config: