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
a95dc025
Commit
a95dc025
authored
Apr 29, 2023
by
AUTOMATIC
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
remove unwanted changes from #8789
parent
f96e6fbd
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
14 deletions
+5
-14
modules/script_callbacks.py
modules/script_callbacks.py
+3
-12
webui.py
webui.py
+2
-2
No files found.
modules/script_callbacks.py
View file @
a95dc025
...
@@ -94,7 +94,6 @@ callback_map = dict(
...
@@ -94,7 +94,6 @@ callback_map = dict(
callbacks_script_unloaded
=
[],
callbacks_script_unloaded
=
[],
callbacks_before_ui
=
[],
callbacks_before_ui
=
[],
callbacks_on_reload
=
[],
callbacks_on_reload
=
[],
callbacks_on_polling
=
[],
)
)
...
@@ -102,6 +101,7 @@ def clear_callbacks():
...
@@ -102,6 +101,7 @@ def clear_callbacks():
for
callback_list
in
callback_map
.
values
():
for
callback_list
in
callback_map
.
values
():
callback_list
.
clear
()
callback_list
.
clear
()
def
app_started_callback
(
demo
:
Optional
[
Blocks
],
app
:
FastAPI
):
def
app_started_callback
(
demo
:
Optional
[
Blocks
],
app
:
FastAPI
):
for
c
in
callback_map
[
'callbacks_app_started'
]:
for
c
in
callback_map
[
'callbacks_app_started'
]:
try
:
try
:
...
@@ -109,14 +109,8 @@ def app_started_callback(demo: Optional[Blocks], app: FastAPI):
...
@@ -109,14 +109,8 @@ def app_started_callback(demo: Optional[Blocks], app: FastAPI):
except
Exception
:
except
Exception
:
report_exception
(
c
,
'app_started_callback'
)
report_exception
(
c
,
'app_started_callback'
)
def
app_polling_callback
(
demo
:
Optional
[
Blocks
],
app
:
FastAPI
):
for
c
in
callback_map
[
'callbacks_on_polling'
]:
try
:
c
.
callback
()
except
Exception
:
report_exception
(
c
,
'callbacks_on_polling'
)
def
app_reload_callback
(
demo
:
Optional
[
Blocks
],
app
:
FastAPI
):
def
app_reload_callback
():
for
c
in
callback_map
[
'callbacks_on_reload'
]:
for
c
in
callback_map
[
'callbacks_on_reload'
]:
try
:
try
:
c
.
callback
()
c
.
callback
()
...
@@ -269,14 +263,11 @@ def on_app_started(callback):
...
@@ -269,14 +263,11 @@ def on_app_started(callback):
add_callback
(
callback_map
[
'callbacks_app_started'
],
callback
)
add_callback
(
callback_map
[
'callbacks_app_started'
],
callback
)
def
on_polling
(
callback
):
"""register a function to be called on each polling of the server."""
add_callback
(
callback_map
[
'callbacks_on_polling'
],
callback
)
def
on_before_reload
(
callback
):
def
on_before_reload
(
callback
):
"""register a function to be called just before the server reloads."""
"""register a function to be called just before the server reloads."""
add_callback
(
callback_map
[
'callbacks_on_reload'
],
callback
)
add_callback
(
callback_map
[
'callbacks_on_reload'
],
callback
)
def
on_model_loaded
(
callback
):
def
on_model_loaded
(
callback
):
"""register a function to be called when the stable diffusion model is created; the model is
"""register a function to be called when the stable diffusion model is created; the model is
passed as an argument; this function is also called when the script is reloaded. """
passed as an argument; this function is also called when the script is reloaded. """
...
...
webui.py
View file @
a95dc025
...
@@ -264,13 +264,13 @@ def create_api(app):
...
@@ -264,13 +264,13 @@ def create_api(app):
def
wait_on_server
(
demo
=
None
):
def
wait_on_server
(
demo
=
None
):
while
1
:
while
1
:
time
.
sleep
(
0.5
)
time
.
sleep
(
0.5
)
modules
.
script_callbacks
.
app_polling_callback
(
None
,
demo
)
if
shared
.
state
.
need_restart
:
if
shared
.
state
.
need_restart
:
modules
.
script_callbacks
.
app_reload_callback
(
None
,
demo
)
shared
.
state
.
need_restart
=
False
shared
.
state
.
need_restart
=
False
time
.
sleep
(
0.5
)
time
.
sleep
(
0.5
)
demo
.
close
()
demo
.
close
()
time
.
sleep
(
0.5
)
time
.
sleep
(
0.5
)
modules
.
script_callbacks
.
app_reload_callback
()
break
break
...
...
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