diff options
| author | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-09-09 08:28:06 +0000 |
|---|---|---|
| committer | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-09-11 10:37:55 +0000 |
| commit | e785402b6acca12108e15224ff80d58817ab3c27 (patch) | |
| tree | 101c52f9e5f9b4915c7809958166b475cadc6e9c /modules/ui_extra_networks_textual_inversion.py | |
| parent | f5959c1c3022c454de22fab749d0f06ab3219868 (diff) | |
| download | stable-diffusion-webui-gfx803-e785402b6acca12108e15224ff80d58817ab3c27.tar.gz stable-diffusion-webui-gfx803-e785402b6acca12108e15224ff80d58817ab3c27.tar.bz2 stable-diffusion-webui-gfx803-e785402b6acca12108e15224ff80d58817ab3c27.zip | |
return nothing if not found
Diffstat (limited to 'modules/ui_extra_networks_textual_inversion.py')
| -rw-r--r-- | modules/ui_extra_networks_textual_inversion.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/modules/ui_extra_networks_textual_inversion.py b/modules/ui_extra_networks_textual_inversion.py index 40ab0aca..71c38fab 100644 --- a/modules/ui_extra_networks_textual_inversion.py +++ b/modules/ui_extra_networks_textual_inversion.py @@ -14,6 +14,8 @@ class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage): def create_item(self, name, index=None, enable_filter=True):
embedding = sd_hijack.model_hijack.embedding_db.word_embeddings.get(name)
+ if embedding is None:
+ return
path, ext = os.path.splitext(embedding.filename)
return {
@@ -31,7 +33,9 @@ class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage): def list_items(self):
names = list(sd_hijack.model_hijack.embedding_db.word_embeddings)
for index, name in enumerate(names):
- yield self.create_item(name, index)
+ item = self.create_item(name, index)
+ if item is not None:
+ yield item
def allowed_directories_for_previews(self):
return list(sd_hijack.model_hijack.embedding_db.embedding_dirs)
|