rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Merge pull request #267 from pi6am/fix/streaming-spaces

Browse Source 
Fix space tokens being dropped from stream preview.
This commit is contained in:
ebolam
2022-10-30 12:27:32 -04:00
committed by GitHub
parent 559ec4d55c 9f0b81742b
commit 1fe41aaceb
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
static/application.js
View File

@@ -79,6 +79,7 @@ var rs_close;
var seqselmenu; var seqselmenu;
var seqselcontents; var seqselcontents;
var stream_preview; var stream_preview;
var stream_preview_text;
var token_prob_container; var token_prob_container;
var storyname = null; var storyname = null;
@@ -2151,6 +2152,7 @@ function endStream() {
if (stream_preview) { if (stream_preview) {
stream_preview.remove(); stream_preview.remove();
stream_preview = null; stream_preview = null;
stream_preview_text = null;
} }
} }
@@ -2390,10 +2392,14 @@ $(document).ready(function(){
if (!stream_preview && streamingEnabled) { if (!stream_preview && streamingEnabled) {
stream_preview = document.createElement("span"); stream_preview = document.createElement("span");
game_text.append(stream_preview); game_text.append(stream_preview);
stream_preview_text = "";
} }
for (const token of msg.data) { for (const token of msg.data) {
if (streamingEnabled) stream_preview.innerText += token.decoded; if (streamingEnabled) {
stream_preview_text += token.decoded;
stream_preview.innerText = stream_preview_text;
}
if (probabilitiesEnabled) { if (probabilitiesEnabled) {
// Probability display // Probability display
@@ -3800,4 +3806,4 @@ function getSelectedOptions(element) {
output.push(item.value); output.push(item.value);
} }
return output; return output;
} }