diff --git a/ldm/invoke/model_cache.py b/ldm/invoke/model_cache.py index 82f46340bb4..7d1654718a3 100644 --- a/ldm/invoke/model_cache.py +++ b/ldm/invoke/model_cache.py @@ -126,18 +126,15 @@ def set_default_model(self,model_name:str) -> None: 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 } ''' models = {} - for name, config in self.config.items(): + for name in self.config: try: - description = config.description + description = self.config[name].description except ConfigAttributeError: description = '' @@ -148,12 +145,10 @@ def list_models(self) -> dict: else: status = 'not loaded' - models.update( - name = { - 'status': status, - 'description': description, - }) - + models[name]={ + 'status' : status, + 'description' : description + } return models def print_models(self) -> None: