diff options
| author | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-09-03 02:56:02 +0000 |
|---|---|---|
| committer | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-09-03 02:56:02 +0000 |
| commit | aab385d01b4311726127397552d791f4d71b7147 (patch) | |
| tree | 0884cec336426d40b7fae698007428f55a9008d3 /modules/ui_extra_networks_textual_inversion.py | |
| parent | d39440bfb9d3b20338fc23a78e6655b1e2f7c1d5 (diff) | |
| download | stable-diffusion-webui-gfx803-aab385d01b4311726127397552d791f4d71b7147.tar.gz stable-diffusion-webui-gfx803-aab385d01b4311726127397552d791f4d71b7147.tar.bz2 stable-diffusion-webui-gfx803-aab385d01b4311726127397552d791f4d71b7147.zip | |
thread safe extra network list_items
Diffstat (limited to 'modules/ui_extra_networks_textual_inversion.py')
| -rw-r--r-- | modules/ui_extra_networks_textual_inversion.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/modules/ui_extra_networks_textual_inversion.py b/modules/ui_extra_networks_textual_inversion.py index 55ef0ea7..d25b45d6 100644 --- a/modules/ui_extra_networks_textual_inversion.py +++ b/modules/ui_extra_networks_textual_inversion.py @@ -29,8 +29,9 @@ class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage): }
def list_items(self):
- for index, name in enumerate(sd_hijack.model_hijack.embedding_db.word_embeddings):
- yield self.create_item(name, index)
+ with self.thread_lock:
+ for index, name in enumerate(sd_hijack.model_hijack.embedding_db.word_embeddings):
+ yield self.create_item(name, index)
def allowed_directories_for_previews(self):
return list(sd_hijack.model_hijack.embedding_db.embedding_dirs)
|