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

Make generation actually work

This commit is contained in:
Cohee 2024-07-04 22:51:47 +03:00
parent e32b0cc223
commit 5da3921907
2 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
public/scripts/extensions/stable-diffusion/index.js
View File

@ -2725,12 +2725,10 @@ async function generateStabilityImage(prompt, negativePrompt) {
}); });
if (!response.ok) { if (!response.ok) {
const errorText = await response.text(); throw new Error(`HTTP ${response.status}: ${response.statusText}`);
throw new Error(`HTTP ${response.status}: ${errorText}`);
} }
const blob = await response.blob(); const base64Image = await response.text();
const base64Image = await getBase64Async(blob);
return { return {
format: IMAGE_FORMAT, format: IMAGE_FORMAT,

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

@ -836,6 +836,7 @@ stability.post('/generate', jsonParser, async (request, response) => {
'Accept': 'image/*', 'Accept': 'image/*',
}, },
body: formData, body: formData,
timeout: 0,
}); });
if (!result.ok) { if (!result.ok) {
@ -845,7 +846,7 @@ stability.post('/generate', jsonParser, async (request, response) => {
} }
const buffer = await result.buffer(); const buffer = await result.buffer();
return response.send(buffer); return response.send(buffer.toString('base64'));
} catch (error) { } catch (error) {
console.log(error); console.log(error);
return response.sendStatus(500); return response.sendStatus(500);