This commit is contained in:
RossAscends
2023-05-16 22:04:19 +09:00
3 changed files with 14 additions and 6 deletions

View File

@@ -229,6 +229,10 @@ async function loadHordeSamplers() {
}
async function loadExtrasSamplers() {
if (!modules.includes('sd')) {
return [];
}
const url = new URL(getApiUrl());
url.pathname = '/api/image/samplers';
const result = await fetch(url, defaultRequestArgs);
@@ -276,6 +280,10 @@ async function loadHordeModels() {
}
async function loadExtrasModels() {
if (!modules.includes('sd')) {
return [];
}
const url = new URL(getApiUrl());
url.pathname = '/api/image/model';
const getCurrentModelResult = await fetch(url, defaultRequestArgs);
@@ -401,11 +409,12 @@ async function generateExtrasImage(prompt) {
}),
});
if (result.ok) {
const data = await result.json();
const base64Image = `data:image/jpeg;base64,${data.image}`;
sendMessage(prompt, base64Image);
} else {
callPopup('Image generation has failed. Please try again.', 'text');
}
}
@@ -431,6 +440,8 @@ async function generateHordeImage(prompt) {
const data = await result.text();
const base64Image = `data:image/webp;base64,${data}`;
sendMessage(prompt, base64Image);
} else {
callPopup('Image generation has failed. Please try again.', 'text');
}
}