Commit aab385d0 authored by w-e-w's avatar w-e-w
Browse files

thread safe extra network list_items

parent d39440bf
Loading
Loading
Loading
Loading
+5 −5
Original line number Diff line number Diff line
@@ -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

+2 −0
Original line number Diff line number Diff line
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
+3 −3
Original line number Diff line number Diff line
@@ -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):
+3 −2
Original line number Diff line number Diff line
@@ -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)

+3 −2
Original line number Diff line number Diff line
@@ -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)