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

refactor(model_manager): avoid copy/paste logic

Browse Source
This commit is contained in:
Kevin Turner 2023-07-30 11:53:12 -07:00
parent 0e48c98330
commit adfd1e52f4
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -382,10 +382,9 @@ class ModelManager(object):
# causing otherwise unreferenced models to be removed from memory # causing otherwise unreferenced models to be removed from memory
self._read_models() self._read_models()
def model_exists(self, model_name: str, base_model: BaseModelType, model_type: ModelType, rescan=False) -> bool: def model_exists(self, model_name: str, base_model: BaseModelType, model_type: ModelType, *, rescan=False) -> bool:
""" """
Given a model name, returns True if it is a valid Given a model name, returns True if it is a valid identifier.
identifier.
""" """
model_key = self.create_key(model_name, base_model, model_type) model_key = self.create_key(model_name, base_model, model_type)
exists = model_key in self.models exists = model_key in self.models
@ -393,7 +392,7 @@ class ModelManager(object):
# if model not found try to find it (maybe file just pasted) # if model not found try to find it (maybe file just pasted)
if rescan and not exists: if rescan and not exists:
self.scan_models_directory(base_model=base_model, model_type=model_type) self.scan_models_directory(base_model=base_model, model_type=model_type)
exists = model_key in self.models exists = self.model_exists(model_name, base_model, model_type, rescan=False)
return exists return exists