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

lint: formatting

Browse Source
This commit is contained in:
Kevin Turner 2023-07-29 20:02:31 -07:00
parent dbfd1bcb5e
commit ff1c40747e
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -258,7 +258,8 @@ from .models import (
ModelConfigBase,
ModelNotFoundException,
InvalidModelException,
DuplicateModelException, ModelBase,
DuplicateModelException,
ModelBase,
)
# We are only starting to number the config file with release 3.
@ -381,13 +382,7 @@ class ModelManager(object):
# causing otherwise unreferenced models to be removed from memory
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
identifier.
@ -540,8 +535,9 @@ class ModelManager(object):
model_class = MODEL_CLASSES[base_model][model_type]
return model_class
def _instantiate(self, model_name: str, base_model: BaseModelType, model_type: ModelType,
submodel_type: SubModelType = None) -> ModelBase:
def _instantiate(
self, model_name: str, base_model: BaseModelType, model_type: ModelType, submodel_type: SubModelType = None
) -> ModelBase:
model_config = self._get_model_config(base_model, model_name, model_type)
model_path, is_submodel_override = self._get_model_path(model_config, submodel_type)
# FIXME: do non-overriden submodels get the right class?