Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Contribute to GitLab
Sign in / Register
Toggle navigation
S
stable-diffusion-webui
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Administrator
stable-diffusion-webui
Commits
e5690d0b
Unverified
Commit
e5690d0b
authored
Nov 12, 2022
by
AUTOMATIC1111
Committed by
GitHub
Nov 12, 2022
Browse files
Options
Browse Files
Download
Plain Diff
Merge pull request #4488 from d8ahazard/ExtensionPreload
Add option to preload extensions
parents
0ab0a50f
cfcadeae
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
26 additions
and
2 deletions
+26
-2
extensions.py
modules/extensions.py
+22
-1
shared.py
modules/shared.py
+4
-1
No files found.
modules/extensions.py
View file @
e5690d0b
import
os
import
os
import
sys
import
sys
import
traceback
import
traceback
from
importlib.machinery
import
SourceFileLoader
import
git
import
git
from
modules
import
paths
,
shared
from
modules
import
paths
,
shared
extensions
=
[]
extensions
=
[]
extensions_dir
=
os
.
path
.
join
(
paths
.
script_path
,
"extensions"
)
extensions_dir
=
os
.
path
.
join
(
paths
.
script_path
,
"extensions"
)
...
@@ -84,3 +84,24 @@ def list_extensions():
...
@@ -84,3 +84,24 @@ def list_extensions():
extension
=
Extension
(
name
=
dirname
,
path
=
path
,
enabled
=
dirname
not
in
shared
.
opts
.
disabled_extensions
)
extension
=
Extension
(
name
=
dirname
,
path
=
path
,
enabled
=
dirname
not
in
shared
.
opts
.
disabled_extensions
)
extensions
.
append
(
extension
)
extensions
.
append
(
extension
)
def
preload_extensions
(
parser
):
if
not
os
.
path
.
isdir
(
extensions_dir
):
return
for
dirname
in
sorted
(
os
.
listdir
(
extensions_dir
)):
path
=
os
.
path
.
join
(
extensions_dir
,
dirname
)
if
not
os
.
path
.
isdir
(
path
):
continue
for
file
in
os
.
listdir
(
path
):
if
"preload.py"
in
file
:
full_file
=
os
.
path
.
join
(
path
,
file
)
print
(
f
"Got preload file: {full_file}"
)
try
:
ext
=
SourceFileLoader
(
"preload"
,
full_file
)
.
load_module
()
parser
=
ext
.
preload
(
parser
)
except
Exception
as
e
:
print
(
f
"Exception preloading script: {e}"
)
return
parser
\ No newline at end of file
modules/shared.py
View file @
e5690d0b
...
@@ -15,7 +15,7 @@ import modules.memmon
...
@@ -15,7 +15,7 @@ import modules.memmon
import
modules.sd_models
import
modules.sd_models
import
modules.styles
import
modules.styles
import
modules.devices
as
devices
import
modules.devices
as
devices
from
modules
import
sd_samplers
,
sd_models
,
localization
,
sd_vae
from
modules
import
sd_samplers
,
sd_models
,
localization
,
sd_vae
,
extensions
from
modules.hypernetworks
import
hypernetwork
from
modules.hypernetworks
import
hypernetwork
from
modules.paths
import
models_path
,
script_path
,
sd_path
from
modules.paths
import
models_path
,
script_path
,
sd_path
...
@@ -91,7 +91,10 @@ parser.add_argument("--tls-keyfile", type=str, help="Partially enables TLS, requ
...
@@ -91,7 +91,10 @@ parser.add_argument("--tls-keyfile", type=str, help="Partially enables TLS, requ
parser
.
add_argument
(
"--tls-certfile"
,
type
=
str
,
help
=
"Partially enables TLS, requires --tls-keyfile to fully function"
,
default
=
None
)
parser
.
add_argument
(
"--tls-certfile"
,
type
=
str
,
help
=
"Partially enables TLS, requires --tls-keyfile to fully function"
,
default
=
None
)
parser
.
add_argument
(
"--server-name"
,
type
=
str
,
help
=
"Sets hostname of server"
,
default
=
None
)
parser
.
add_argument
(
"--server-name"
,
type
=
str
,
help
=
"Sets hostname of server"
,
default
=
None
)
extensions
.
preload_extensions
(
parser
)
cmd_opts
=
parser
.
parse_args
()
cmd_opts
=
parser
.
parse_args
()
restricted_opts
=
{
restricted_opts
=
{
"samples_filename_pattern"
,
"samples_filename_pattern"
,
"directories_filename_pattern"
,
"directories_filename_pattern"
,
...
...
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