networks avoid slow yield

Signed-off-by: Vladimir Mandic <mandic00@live.com>

modules/ui_extra_networks_checkpoints.py

@@ -145,7 +145,6 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
                 "tags": v.get('tags', []),
             }
             models.append(model)
-            # yield model
         log.debug(f'Networks: type="reference" {count}')
         return models
 

modules/ui_extra_networks_history.py

@@ -19,9 +19,10 @@ class ExtraNetworksPageHistory(ui_extra_networks.ExtraNetworksPage):
 
     def list_items(self):
         # log.trace('History list')
+        records = []
         for item in shared.history.latents:
             title = ', '.join(list(set(item.ops))) + '<br>' + item.name
-            yield {
+            record = {
                 "type": 'History',
                 "name": title,
                 "preview": item.preview,
@@ -31,6 +32,8 @@ class ExtraNetworksPageHistory(ui_extra_networks.ExtraNetworksPage):
                 # "description": item.info,
                 "onclick": '"' + html.escape(f"""return selectHistory({json.dumps(item.name)})""") + '"',
             }
+            records.append(record)
+        return records
 
     def find_description(self, path, info=None):
         name = path.split('<br>')[-1]

modules/ui_extra_networks_styles.py

@@ -137,6 +137,5 @@ class ExtraNetworkStyles(extra_networks.ExtraNetwork):
                     p.negative_prompts = [styles.merge_prompts(style.negative_prompt, prompt) for prompt in p.negative_prompts]
                     styles.apply_styles_to_extra(p, style)
 
-
     def deactivate(self, p, force=False):
         pass

modules/ui_extra_networks_unet.py

@@ -13,6 +13,7 @@ class ExtraNetworksPageUNets(ui_extra_networks.ExtraNetworksPage):
         return sd_unet.refresh_unet_list()
 
     def list_items(self):
+        results = []
         for name, filename in sd_unet.unet_dict.items():
             try:
                 size, mtime = modelstats.stat(filename)
@@ -35,9 +36,10 @@ class ExtraNetworksPageUNets(ui_extra_networks.ExtraNetworksPage):
                     "description": self.find_description(filename, info),
                     "version": version.get("baseModel", "N/A") if info else "N/A",
                 }
-                yield record
+                results.append(record)
             except Exception as e:
                 log.debug(f'Networks error: type=unet file="{filename}" {e}')
+        return results
 
     def allowed_directories_for_previews(self):
         return [v for v in [shared.opts.unet_dir] if v is not None]

modules/ui_extra_networks_vae.py

@@ -10,9 +10,10 @@ class ExtraNetworksPageVAEs(ui_extra_networks.ExtraNetworksPage):
         super().__init__('VAE')
 
     def refresh(self):
-        shared.refresh_vaes()
+        return shared.refresh_vaes()
 
     def list_items(self):
+        records = []
         for name, filename in sd_vae.vae_dict.items():
             try:
                 size, mtime = modelstats.stat(filename)
@@ -35,9 +36,10 @@ class ExtraNetworksPageVAEs(ui_extra_networks.ExtraNetworksPage):
                     "description": self.find_description(filename, info),
                     "version": version.get("baseModel", "N/A") if info else "N/A",
                 }
-                yield record
+                records.append(record)
             except Exception as e:
                 log.debug(f'Networks error: type=vae file="{filename}" {e}')
+        return records
 
     def allowed_directories_for_previews(self):
         return [v for v in [shared.opts.vae_dir] if v is not None]