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