aboutsummaryrefslogtreecommitdiff
path: root/modules/api
diff options
context:
space:
mode:
authorRiccardo Giovanetti <29801031+Harvester62@users.noreply.github.com>2022-11-01 16:36:27 +0100
committerGitHub <noreply@github.com>2022-11-01 16:36:27 +0100
commita3eab2f71e53c56b2de767550250a93749c6630b (patch)
tree6ce0860a7edd29dbeaa8c6085d3437b257d9ff1d /modules/api
parent7b6a412709ca229688c3b4dd1ab793d39acf97ef (diff)
parentfb3931400672119f5c842becdf996315dd7bc855 (diff)
Merge branch 'AUTOMATIC1111:master' into Italian
Diffstat (limited to 'modules/api')
-rw-r--r--modules/api/api.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/modules/api/api.py b/modules/api/api.py
index 6c06d449..e702c9c0 100644
--- a/modules/api/api.py
+++ b/modules/api/api.py
@@ -40,6 +40,7 @@ class Api:
self.app.add_api_route("/sdapi/v1/extra-batch-images", self.extras_batch_images_api, methods=["POST"], response_model=ExtrasBatchImagesResponse)
self.app.add_api_route("/sdapi/v1/png-info", self.pnginfoapi, methods=["POST"], response_model=PNGInfoResponse)
self.app.add_api_route("/sdapi/v1/progress", self.progressapi, methods=["GET"], response_model=ProgressResponse)
+ self.app.add_api_route("/sdapi/v1/interrupt", self.interruptapi, methods=["POST"])
def text2imgapi(self, txt2imgreq: StableDiffusionTxt2ImgProcessingAPI):
sampler_index = sampler_to_index(txt2imgreq.sampler_index)
@@ -176,6 +177,11 @@ class Api:
return ProgressResponse(progress=progress, eta_relative=eta_relative, state=shared.state.dict(), current_image=current_image)
+ def interruptapi(self):
+ shared.state.interrupt()
+
+ return {}
+
def launch(self, server_name, port):
self.app.include_router(self.router)
uvicorn.run(self.app, host=server_name, port=port)