diff --git a/ldm/invoke/model_cache.py b/ldm/invoke/model_cache.py index 3bb1b7e928a..6d4e592e447 100644 --- a/ldm/invoke/model_cache.py +++ b/ldm/invoke/model_cache.py @@ -134,12 +134,14 @@ def list_models(self) -> dict: } ''' models = {} - for name, config in self.config.items(): + for config_entry in self.config.items(): + (name, config) = config_entry + try: description = config.description except ConfigAttributeError: description = '' - + if self.current_model == name: status = 'active' elif name in self.models: @@ -147,12 +149,9 @@ def list_models(self) -> dict: else: status = 'not loaded' - models = models.update( - name = { - 'status': status, - 'description': description, - }) - + model_details = {name: {'status': status, 'description': description,}} + models.update(model_details) + return models def print_models(self) -> None: