diff --git a/public/scripts/extensions/stable-diffusion/index.js b/public/scripts/extensions/stable-diffusion/index.js index ed7161140..4a5667c10 100644 --- a/public/scripts/extensions/stable-diffusion/index.js +++ b/public/scripts/extensions/stable-diffusion/index.js @@ -2725,12 +2725,10 @@ async function generateStabilityImage(prompt, negativePrompt) { }); if (!response.ok) { - const errorText = await response.text(); - throw new Error(`HTTP ${response.status}: ${errorText}`); + throw new Error(`HTTP ${response.status}: ${response.statusText}`); } - const blob = await response.blob(); - const base64Image = await getBase64Async(blob); + const base64Image = await response.text(); return { format: IMAGE_FORMAT, diff --git a/src/endpoints/stable-diffusion.js b/src/endpoints/stable-diffusion.js index dde53bc17..d5ba855b2 100644 --- a/src/endpoints/stable-diffusion.js +++ b/src/endpoints/stable-diffusion.js @@ -836,6 +836,7 @@ stability.post('/generate', jsonParser, async (request, response) => { 'Accept': 'image/*', }, body: formData, + timeout: 0, }); if (!result.ok) { @@ -845,7 +846,7 @@ stability.post('/generate', jsonParser, async (request, response) => { } const buffer = await result.buffer(); - return response.send(buffer); + return response.send(buffer.toString('base64')); } catch (error) { console.log(error); return response.sendStatus(500);