horde worker model in message tooltip

This commit is contained in:
RossAscends
2023-05-18 18:50:18 +09:00
parent 780e00938f
commit 8d11d5d8d5

View File

@ -102,7 +102,7 @@ async function generateHorde(prompt, params) {
const response = await fetch("/generate_horde", { const response = await fetch("/generate_horde", {
method: 'POST', method: 'POST',
headers: { headers: {
...getRequestHeaders(), ...getRequestHeaders(),
"Client-Agent": CLIENT_VERSION, "Client-Agent": CLIENT_VERSION,
}, },
body: JSON.stringify(payload) body: JSON.stringify(payload)
@ -129,9 +129,10 @@ async function generateHorde(prompt, params) {
setGenerationProgress(100); setGenerationProgress(100);
const generatedText = statusCheckJson.generations[0].text; const generatedText = statusCheckJson.generations[0].text;
const WorkerName = statusCheckJson.generations[0].worker_name; const WorkerName = statusCheckJson.generations[0].worker_name;
const WorkerModel = statusCheckJson.generations[0].model;
console.log(generatedText); console.log(generatedText);
console.log(`Generated by Horde Worker: ${WorkerName}`); console.log(`Generated by Horde Worker: ${WorkerName} [${WorkerModel}]`);
return { text: generatedText, workerName: `Generated by Horde worker: ${WorkerName}` }; return { text: generatedText, workerName: `Generated by Horde worker: ${WorkerName} [${WorkerModel}]` };
} }
else if (!queue_position_first) { else if (!queue_position_first) {
queue_position_first = statusCheckJson.queue_position; queue_position_first = statusCheckJson.queue_position;