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

test(model_management): add a couple tests for _get_model_path

Browse Source
This commit is contained in:
Kevin Turner
2023-08-05 15:22:23 -07:00
parent 65ed224bfc
commit 44bf308192
6 changed files with 53 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -533,7 +533,7 @@ class ModelManager(object):
model_path = self.resolve_model_path(model_path)
return model_path, is_submodel_override
def _get_model_config(self, base_model, model_name, model_type) -> ModelConfigBase:
def _get_model_config(self, base_model: BaseModelType, model_name: str, model_type: ModelType) -> ModelConfigBase:
"""Get a model's config object."""
model_key = self.create_key(model_name, base_model, model_type)
try: