diff --git a/ldm/invoke/model_cache.py b/ldm/invoke/model_cache.py index 645a6fd4da..3bb1b7e928 100644 --- a/ldm/invoke/model_cache.py +++ b/ldm/invoke/model_cache.py @@ -125,15 +125,18 @@ class ModelCache(object): def list_models(self) -> dict: ''' Return a dict of models in the format: - { model_name1: {'status': ('active'|'cached'|'not loaded'), - 'description': description, - }, - model_name2: { etc } + { + model_name1: { + 'status': ('active'|'cached'|'not loaded'), + 'description': description, + }, + model_name2: { etc }, + } ''' - result = dict() - for name in self.config: + models = {} + for name, config in self.config.items(): try: - description = self.config[name].description + description = config.description except ConfigAttributeError: description = '' @@ -144,11 +147,13 @@ class ModelCache(object): else: status = 'not loaded' - result[name]={ - 'status' : status, - 'description' : description - } - return result + models = models.update( + name = { + 'status': status, + 'description': description, + }) + + return models def print_models(self) -> None: '''