diff options
| author | ctwrs <> | 2023-02-01 20:58:09 +0000 |
|---|---|---|
| committer | ctwrs <> | 2023-02-01 21:28:39 +0000 |
| commit | 92bae77b88fd90743eebec69ca7af1ee1c6e40f2 (patch) | |
| tree | cb593502e5a8a0024f242faf1aadc740b4a592db /modules/ui_extra_networks.py | |
| parent | 226d840e84c5f306350b0681945989b86760e616 (diff) | |
| download | stable-diffusion-webui-gfx803-92bae77b88fd90743eebec69ca7af1ee1c6e40f2.tar.gz stable-diffusion-webui-gfx803-92bae77b88fd90743eebec69ca7af1ee1c6e40f2.tar.bz2 stable-diffusion-webui-gfx803-92bae77b88fd90743eebec69ca7af1ee1c6e40f2.zip | |
Add .jpg to allowed thumb formats
Diffstat (limited to 'modules/ui_extra_networks.py')
| -rw-r--r-- | modules/ui_extra_networks.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py index 83367968..95b30f4a 100644 --- a/modules/ui_extra_networks.py +++ b/modules/ui_extra_networks.py @@ -29,8 +29,9 @@ def add_pages_to_demo(app): if not any([Path(x).resolve() in Path(filename).resolve().parents for x in allowed_dirs]):
raise ValueError(f"File cannot be fetched: {filename}. Must be in one of directories registered by extra pages.")
- if os.path.splitext(filename)[1].lower() != ".png":
- raise ValueError(f"File cannot be fetched: {filename}. Only png.")
+ ext = os.path.splitext(filename)[1].lower()
+ if ext not in (".png", ".jpg"):
+ raise ValueError(f"File cannot be fetched: {filename}. Only png and jpg.")
# would profit from returning 304
return FileResponse(filename, headers={"Accept-Ranges": "bytes"})
|