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

added a wrapper model_manager_service and model events

Browse Source
This commit is contained in:
Lincoln Stein
2023-05-11 00:09:19 -04:00
parent fa6a580452
commit 4627910c5d
6 changed files with 495 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -479,10 +479,11 @@ class ModelManager(object):
line = f"\033[1m{line}\033[0m"
print(line)
def del_model(self, model_name: str, delete_files: bool = False) -> None:
def del_model(self, model_name: str, model_type: SDModelType.diffusers, delete_files: bool = False):
"""
Delete the named model.
"""
model_name = self._disambiguate_name(model_name, model_type)
omega = self.config
if model_name not in omega:
self.logger.error(f"Unknown model {model_name}")