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

make relative model paths work in model manager

Browse Source
This commit is contained in:
Lincoln Stein
2023-06-23 10:52:59 -04:00
parent a910403003
commit 56bd873d7a
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -384,7 +384,8 @@ class ModelManager(object):
# if it known model check that target path exists (if manualy deleted) # if it known model check that target path exists (if manualy deleted)
else: else:
# logic repeated twice(in rescan too) any way to optimize? # logic repeated twice(in rescan too) any way to optimize?
if not os.path.exists(self.models[model_key].path): model_path = self.globals.root_path / self.models[model_key].path
if not model_path.exists():
if model_class.save_to_config: if model_class.save_to_config:
self.models[model_key].error = ModelError.NotFound self.models[model_key].error = ModelError.NotFound
raise Exception(f"Files for model \"{model_key}\" not found") raise Exception(f"Files for model \"{model_key}\" not found")
@ -395,13 +396,13 @@ class ModelManager(object):
# reset model errors? # reset model errors?
model_config = self.models[model_key] model_config = self.models[model_key]
# /models/{base_model}/{model_type}/{name}.ckpt or .safentesors # /models/{base_model}/{model_type}/{name}.ckpt or .safentesors
# /models/{base_model}/{model_type}/{name}/ # /models/{base_model}/{model_type}/{name}/
model_path = model_config.path # massage relative paths into absolute ones
model_path = model_path or self.globals.root_path / model_config.path
model_config.path = model_path
# vae/movq override # vae/movq override
# TODO: # TODO:
@ -415,7 +416,7 @@ class ModelManager(object):
# TODO: path # TODO: path
# TODO: is it accurate to use path as id # TODO: is it accurate to use path as id
dst_convert_path = self.globals.models_dir / ".cache" / hashlib.md5(model_path.encode()).hexdigest() dst_convert_path = self.globals.models_dir / ".cache" / hashlib.md5(str(model_path).encode()).hexdigest()
model_path = model_class.convert_if_required( model_path = model_class.convert_if_required(
base_model=base_model, base_model=base_model,
model_path=model_path, model_path=model_path,