Skip to content

Commit d51a5d6

Browse files
Merge pull request #4025 from evshiron/feat/interrupt-api-master
prototype interrupt api
2 parents efd20a4 + adaa699 commit d51a5d6

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

modules/api/api.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ def __init__(self, app, queue_lock):
4040
self.app.add_api_route("/sdapi/v1/extra-batch-images", self.extras_batch_images_api, methods=["POST"], response_model=ExtrasBatchImagesResponse)
4141
self.app.add_api_route("/sdapi/v1/png-info", self.pnginfoapi, methods=["POST"], response_model=PNGInfoResponse)
4242
self.app.add_api_route("/sdapi/v1/progress", self.progressapi, methods=["GET"], response_model=ProgressResponse)
43+
self.app.add_api_route("/sdapi/v1/interrupt", self.interruptapi, methods=["POST"])
4344

4445
def text2imgapi(self, txt2imgreq: StableDiffusionTxt2ImgProcessingAPI):
4546
sampler_index = sampler_to_index(txt2imgreq.sampler_index)
@@ -176,6 +177,11 @@ def progressapi(self, req: ProgressRequest = Depends()):
176177

177178
return ProgressResponse(progress=progress, eta_relative=eta_relative, state=shared.state.dict(), current_image=current_image)
178179

180+
def interruptapi(self):
181+
shared.state.interrupt()
182+
183+
return {}
184+
179185
def launch(self, server_name, port):
180186
self.app.include_router(self.router)
181187
uvicorn.run(self.app, host=server_name, port=port)

0 commit comments

Comments
 (0)