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 the docstring more readable and improve the list_models logic (fixes #1539) (#1594)

Browse Source
This commit is contained in:
Damian Stewart 2022-11-28 21:20:56 +01:00 committed by GitHub
commit 281a2e3ecb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
ldm/invoke/model_cache.py
View File

@ -125,15 +125,18 @@ class ModelCache(object):
def list_models(self) -> dict: def list_models(self) -> dict:
''' '''
Return a dict of models in the format: Return a dict of models in the format:
{ model_name1: {'status': ('active'|'cached'|'not loaded'), {
model_name1: {
'status': ('active'|'cached'|'not loaded'),
'description': description, 'description': description,
}, },
model_name2: { etc } model_name2: { etc },
}
''' '''
result = dict() models = {}
for name in self.config: for name, config in self.config.items():
try: try:
description = self.config[name].description description = config.description
except ConfigAttributeError: except ConfigAttributeError:
description = '<no description>' description = '<no description>'
@ -144,11 +147,13 @@ class ModelCache(object):
else: else:
status = 'not loaded' status = 'not loaded'
result[name]={ models.update(
'status' : status, name = {
'description' : description 'status': status,
} 'description': description,
return result })
return models
def print_models(self) -> None: def print_models(self) -> None:
''' '''