diff options
| author | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-06-12 09:15:27 +0000 |
|---|---|---|
| committer | w-e-w <40751091+w-e-w@users.noreply.github.com> | 2023-06-12 09:15:27 +0000 |
| commit | b9664ab6154818680ee25920e229b808a3cdec68 (patch) | |
| tree | c1273af6374e8e098f252ee901abbfb1907788e2 | |
| parent | 9142be0a0d8cea37cf1ae86c17fc7dcb161d9a42 (diff) | |
| download | stable-diffusion-webui-gfx803-b9664ab6154818680ee25920e229b808a3cdec68.tar.gz stable-diffusion-webui-gfx803-b9664ab6154818680ee25920e229b808a3cdec68.tar.bz2 stable-diffusion-webui-gfx803-b9664ab6154818680ee25920e229b808a3cdec68.zip | |
move _stop route to api
| -rw-r--r-- | modules/api/api.py | 11 | ||||
| -rw-r--r-- | webui.py | 7 |
2 files changed, 9 insertions, 9 deletions
diff --git a/modules/api/api.py b/modules/api/api.py index 317c809f..cb1cde78 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -208,8 +208,11 @@ class Api: self.add_api_route("/sdapi/v1/reload-checkpoint", self.reloadapi, methods=["POST"]) self.add_api_route("/sdapi/v1/scripts", self.get_scripts_list, methods=["GET"], response_model=models.ScriptsList) self.add_api_route("/sdapi/v1/script-info", self.get_script_info, methods=["GET"], response_model=List[models.ScriptInfo]) - self.add_api_route("/sdapi/v1/quit-webui", self.quit_webui, methods=["POST"]) - self.add_api_route("/sdapi/v1/restart-webui", self.restart_webui, methods=["POST"]) + + if shared.cmd_opts.add_stop_route: + self.add_api_route("/sdapi/v1/quit-webui", self.quit_webui, methods=["POST"]) + self.add_api_route("/sdapi/v1/restart-webui", self.restart_webui, methods=["POST"]) + self.add_api_route("/_stop", self.stop_route, methods=["POST"]) self.default_script_arg_txt2img = [] self.default_script_arg_img2img = [] @@ -724,3 +727,7 @@ class Api: def restart_webui(self): if restart.is_restartable(): restart.restart_program() + + def stop_route(request): + shared.state.server_command = "stop" + return Response("Stopping.") @@ -362,11 +362,6 @@ def api_only(): api.launch(server_name="0.0.0.0" if cmd_opts.listen else "127.0.0.1", port=cmd_opts.port if cmd_opts.port else 7861)
-def stop_route(request):
- shared.state.server_command = "stop"
- return Response("Stopping.")
-
-
def webui():
launch_api = cmd_opts.api
initialize()
@@ -404,8 +399,6 @@ def webui(): "redoc_url": "/redoc",
},
)
- if cmd_opts.add_stop_route:
- app.add_route("/_stop", stop_route, methods=["POST"])
# after initial launch, disable --autolaunch for subsequent restarts
cmd_opts.autolaunch = False
|