thread safe extra network list_items

extensions-builtin/Lora/ui_extra_networks_lora.py

@@ -66,9 +66,9 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
         return item
 
     def list_items(self):
+        with self.thread_lock:
             for index, name in enumerate(networks.available_networks):
                 item = self.create_item(name, index)
-
                 if item is not None:
                     yield item
 

modules/ui_extra_networks.py

@@ -1,6 +1,7 @@
 import os.path
 import urllib.parse
 from pathlib import Path
+from threading import Lock
 
 from modules import shared, ui_extra_networks_user_metadata, errors, extra_networks
 from modules.images import read_info_from_image, save_image_with_geninfo
@@ -94,6 +95,7 @@ class ExtraNetworksPage:
         self.allow_negative_prompt = False
         self.metadata = {}
         self.items = {}
+        self.thread_lock = Lock()
 
     def refresh(self):
         pass

modules/ui_extra_networks_checkpoints.py

@@ -30,8 +30,8 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
         }
 
     def list_items(self):
-        names = list(sd_models.checkpoints_list)
-        for index, name in enumerate(names):
+        with self.thread_lock:
+            for index, name in enumerate(sd_models.checkpoints_list):
                 yield self.create_item(name, index)
 
     def allowed_directories_for_previews(self):

modules/ui_extra_networks_hypernets.py

@@ -31,6 +31,7 @@ class ExtraNetworksPageHypernetworks(ui_extra_networks.ExtraNetworksPage):
         }
 
     def list_items(self):
+        with self.thread_lock:
             for index, name in enumerate(shared.hypernetworks):
                 yield self.create_item(name, index)
 

modules/ui_extra_networks_textual_inversion.py

@@ -29,6 +29,7 @@ class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage):
         }
 
     def list_items(self):
+        with self.thread_lock:
             for index, name in enumerate(sd_hijack.model_hijack.embedding_db.word_embeddings):
                 yield self.create_item(name, index)