Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
S
Stable Diffusion Webui
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Locked Files
Issues
0
Issues
0
List
Boards
Labels
Service Desk
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Security & Compliance
Security & Compliance
Dependency List
License Compliance
Packages
Packages
List
Container Registry
Analytics
Analytics
CI / CD
Code Review
Insights
Issues
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
novelai-storage
Stable Diffusion Webui
Commits
45881703
Commit
45881703
authored
Sep 07, 2023
by
w-e-w
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
consolidated allowed preview formats
parent
d39440bf
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
6 deletions
+7
-6
modules/ui_extra_networks.py
modules/ui_extra_networks.py
+7
-6
No files found.
modules/ui_extra_networks.py
View file @
45881703
...
...
@@ -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
):
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment