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

prevent crash on rename operation on models in models directory

Browse Source
This commit is contained in:
Lincoln Stein 2023-07-17 07:50:06 -04:00
parent 84a13ff8e1
commit 0ea8d3c30c
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -639,7 +639,6 @@ class ModelManager(object):
model_info(). model_info().
""" """
# relativize paths as they go in - this makes it easier to move the root directory around # relativize paths as they go in - this makes it easier to move the root directory around
self.logger.debug(model_attributes)
if path := model_attributes.get('path'): if path := model_attributes.get('path'):
if Path(path).is_relative_to(self.app_config.root_path): if Path(path).is_relative_to(self.app_config.root_path):
model_attributes['path'] = str(Path(path).relative_to(self.app_config.root_path)) model_attributes['path'] = str(Path(path).relative_to(self.app_config.root_path))
@ -708,7 +707,7 @@ class ModelManager(object):
# if this is a model file/directory that we manage ourselves, we need to move it # if this is a model file/directory that we manage ourselves, we need to move it
if old_path.is_relative_to(self.app_config.models_path): if old_path.is_relative_to(self.app_config.models_path):
new_path = self.app_config.root_path / 'models' / new_base.value / model_type.value / new_name new_path = self.app_config.root_path / 'models' / BaseModelType(new_base).value / ModelType(model_type).value / new_name
move(old_path, new_path) move(old_path, new_path)
model_cfg.path = str(new_path.relative_to(self.app_config.root_path)) model_cfg.path = str(new_path.relative_to(self.app_config.root_path))