shared.state.interrupt when restart is requested

3 years ago · bbdbbd36ed
parent ef40e4cd4d
commit bbdbbd36ed
1 changed files with 1 additions and 0 deletions
--- a/modules/ui.py
+++ b/modules/ui.py
@ -1210,6 +1210,7 @@ def create_ui(wrap_gradio_gpu_call):
        )

        def request_restart():
+            shared.state.interrupt()
            settings_interface.gradio_ref.do_restart = True

        restart_gradio.click(