diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-09-30 07:09:45 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-30 07:09:45 +0000 |
| commit | 416fbde7262a0ecc698647a8bb320fb01bdba388 (patch) | |
| tree | 4b751a38c41633b40869e717dcee74f0d6506027 /modules/ui_extra_networks.py | |
| parent | 951842d7858f94d5d49999f57f37df8e1991c4ae (diff) | |
| parent | 45881703c5b1c0499406a76fa49ec7bd408a4898 (diff) | |
| download | stable-diffusion-webui-gfx803-416fbde7262a0ecc698647a8bb320fb01bdba388.tar.gz stable-diffusion-webui-gfx803-416fbde7262a0ecc698647a8bb320fb01bdba388.tar.bz2 stable-diffusion-webui-gfx803-416fbde7262a0ecc698647a8bb320fb01bdba388.zip | |
Merge pull request #13121 from AUTOMATIC1111/consolidated-allowed-preview-formats
Consolidated allowed preview formats, Fix extra network `.gif` not woking as preview
Diffstat (limited to 'modules/ui_extra_networks.py')
| -rw-r--r-- | modules/ui_extra_networks.py | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py index 3eee371b..d8c31142 100644 --- a/modules/ui_extra_networks.py +++ b/modules/ui_extra_networks.py @@ -15,6 +15,11 @@ from modules.ui_components import ToolButton extra_pages = []
allowed_dirs = set()
+allowed_preview_extensions = ["png", "jpg", "jpeg", "webp", "gif"]
+if shared.opts.samples_format not in allowed_preview_extensions:
+ allowed_preview_extensions.append(shared.opts.samples_format)
+allowed_preview_extensions_dot = ['.' + extension for extension in allowed_preview_extensions]
+
def register_page(page):
"""registers extra networks page for the UI; recommend doing it in on_before_ui() callback for extensions"""
@@ -34,7 +39,7 @@ def fetch_file(filename: str = ""): raise ValueError(f"File cannot be fetched: {filename}. Must be in one of directories registered by extra pages.")
ext = os.path.splitext(filename)[1].lower()
- if ext not in (".png", ".jpg", ".jpeg", ".webp", ".gif"):
+ if ext not in allowed_preview_extensions_dot:
raise ValueError(f"File cannot be fetched: {filename}. Only png, jpg, webp, and gif.")
# would profit from returning 304
@@ -273,11 +278,7 @@ class ExtraNetworksPage: Find a preview PNG for a given path (without extension) and call link_preview on it.
"""
- preview_extensions = ["png", "jpg", "jpeg", "webp"]
- if shared.opts.samples_format not in preview_extensions:
- preview_extensions.append(shared.opts.samples_format)
-
- potential_files = sum([[path + "." + ext, path + ".preview." + ext] for ext in preview_extensions], [])
+ potential_files = sum([[path + "." + ext, path + ".preview." + ext] for ext in allowed_preview_extensions], [])
for file in potential_files:
if os.path.isfile(file):
|