michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

support three extensions for preview instead of one: png, jpg, webp

Browse source
This commit is contained in:
AUTOMATIC 2023-03-11 15:33:24 +03:00
parent ce68ab8d0d
commit 9320139bd8
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
modules/ui_extra_networks.py
View file

@ -142,7 +142,11 @@ class ExtraNetworksPage:
"""
Find a preview PNG for a given path (without extension) and call link_preview on it.
"""
for file in [path + ".png", path + ".preview.png"]:
preview_extensions = ["png", "jpg", "webp"]
potential_files = sum([[path + "." + ext, path + ".preview." + ext] for ext in preview_extensions], [])
for file in potential_files:
if os.path.isfile(file):
return self.link_preview(file)