rixty
/
SillyTavern
mirror of https://github.com/SillyTavern/SillyTavern.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Interrupt AUTO1111 gens on cancel

This commit is contained in:
Cohee 2024-07-19 00:01:24 +03:00
parent 739752ccf3
commit 7e4abaabff
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/endpoints/stable-diffusion.js
View File

@ -323,6 +323,17 @@ router.post('/generate', jsonParser, async (request, response) => {
const url = new URL(request.body.url);
url.pathname = '/sdapi/v1/txt2img';
const controller = new AbortController();
request.socket.removeAllListeners('close');
request.socket.on('close', function () {
if (!response.writableEnded) {
const url = new URL(request.body.url);
url.pathname = '/sdapi/v1/interrupt';
fetch(url, { method: 'POST', headers: { 'Authorization': getBasicAuthHeader(request.body.auth) } });
}
controller.abort();
});
const result = await fetch(url, {
method: 'POST',
body: JSON.stringify(request.body),
@ -331,6 +342,8 @@ router.post('/generate', jsonParser, async (request, response) => {
'Authorization': getBasicAuthHeader(request.body.auth),
},
timeout: 0,
// @ts-ignore
signal: controller.signal,
});
if (!result.ok) {