StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Host busy thread, check for reload

Browse Source
This commit is contained in:
DepFA 2022-10-01 18:31:58 +01:00 committed by GitHub
parent 4c2478a68a
commit 95f35d04ab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 31 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
webui.py
View File

@ -86,6 +86,8 @@ def webui():
signal.signal(signal.SIGINT, sigint_handler)
while 1:
demo = modules.ui.create_ui(
txt2img=wrap_gradio_gpu_call(modules.txt2img.txt2img),
img2img=wrap_gradio_gpu_call(modules.img2img.img2img),
@ -94,6 +96,7 @@ def webui():
run_modelmerger=modules.extras.run_modelmerger
)
demo.launch(
share=cmd_opts.share,
server_name="0.0.0.0" if cmd_opts.listen else None,
@ -101,8 +104,21 @@ def webui():
debug=cmd_opts.gradio_debug,
auth=[tuple(cred.split(':')) for cred in cmd_opts.gradio_auth.strip('"').split(',')] if cmd_opts.gradio_auth else None,
inbrowser=cmd_opts.autolaunch,
prevent_thread_lock=True
)
while 1:
time.sleep(0.5)
if getattr(demo,'do_restart',False):
time.sleep(0.5)
demo.close()
time.sleep(0.5)
break
print('Reloading Scripts')
modules.scripts.reload_scripts(os.path.join(script_path, "scripts"))
print('Restarting Gradio')
if __name__ == "__main__":
webui()