|
|
|
@ -89,38 +89,38 @@ def webui():
|
|
|
|
|
|
|
|
|
|
|
|
while 1:
|
|
|
|
while 1:
|
|
|
|
|
|
|
|
|
|
|
|
demo = modules.ui.create_ui(
|
|
|
|
demo = modules.ui.create_ui(
|
|
|
|
txt2img=wrap_gradio_gpu_call(modules.txt2img.txt2img),
|
|
|
|
txt2img=wrap_gradio_gpu_call(modules.txt2img.txt2img),
|
|
|
|
img2img=wrap_gradio_gpu_call(modules.img2img.img2img),
|
|
|
|
img2img=wrap_gradio_gpu_call(modules.img2img.img2img),
|
|
|
|
run_extras=wrap_gradio_gpu_call(modules.extras.run_extras),
|
|
|
|
run_extras=wrap_gradio_gpu_call(modules.extras.run_extras),
|
|
|
|
run_pnginfo=modules.extras.run_pnginfo,
|
|
|
|
run_pnginfo=modules.extras.run_pnginfo,
|
|
|
|
run_modelmerger=modules.extras.run_modelmerger
|
|
|
|
run_modelmerger=modules.extras.run_modelmerger
|
|
|
|
)
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
demo.launch(
|
|
|
|
demo.launch(
|
|
|
|
share=cmd_opts.share,
|
|
|
|
share=cmd_opts.share,
|
|
|
|
server_name="0.0.0.0" if cmd_opts.listen else None,
|
|
|
|
server_name="0.0.0.0" if cmd_opts.listen else None,
|
|
|
|
server_port=cmd_opts.port,
|
|
|
|
server_port=cmd_opts.port,
|
|
|
|
debug=cmd_opts.gradio_debug,
|
|
|
|
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,
|
|
|
|
auth=[tuple(cred.split(':')) for cred in cmd_opts.gradio_auth.strip('"').split(',')] if cmd_opts.gradio_auth else None,
|
|
|
|
inbrowser=cmd_opts.autolaunch,
|
|
|
|
inbrowser=cmd_opts.autolaunch,
|
|
|
|
prevent_thread_lock=True
|
|
|
|
prevent_thread_lock=True
|
|
|
|
)
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
while 1:
|
|
|
|
while 1:
|
|
|
|
time.sleep(0.5)
|
|
|
|
time.sleep(0.5)
|
|
|
|
if getattr(demo,'do_restart',False):
|
|
|
|
if getattr(demo,'do_restart',False):
|
|
|
|
time.sleep(0.5)
|
|
|
|
time.sleep(0.5)
|
|
|
|
demo.close()
|
|
|
|
demo.close()
|
|
|
|
time.sleep(0.5)
|
|
|
|
time.sleep(0.5)
|
|
|
|
break
|
|
|
|
break
|
|
|
|
|
|
|
|
|
|
|
|
print('Reloading Custom Scripts')
|
|
|
|
print('Reloading Custom Scripts')
|
|
|
|
modules.scripts.reload_scripts(os.path.join(script_path, "scripts"))
|
|
|
|
modules.scripts.reload_scripts(os.path.join(script_path, "scripts"))
|
|
|
|
print('Reloading modules: modules.ui')
|
|
|
|
print('Reloading modules: modules.ui')
|
|
|
|
importlib.reload(modules.ui)
|
|
|
|
importlib.reload(modules.ui)
|
|
|
|
print('Restarting Gradio')
|
|
|
|
print('Restarting Gradio')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
if __name__ == "__main__":
|
|
|
|
|