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

Merge pull request #3002 from 50h100a/streamprobs

Browse Source 
Correctly view token probabilities when 'Continue'-ing a response.
This commit is contained in:
Cohee
2024-10-19 13:14:44 +03:00
committed by GitHub
parent 9cc8830dd3 afde6e3f45
commit 0a79d55983
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
public/css/logprobs.css
View File

@@ -56,6 +56,7 @@
user-select: none; user-select: none;
height: 100%; height: 100%;
overflow-y: auto; overflow-y: auto;
word-break: break-all;
} }
.logprobs_empty_state { .logprobs_empty_state {

13
public/script.js
View File

@@ -3194,7 +3194,8 @@ class StreamingProcessor {
} }
onStopStreaming() { onStopStreaming() {
this.onErrorStreaming(); this.abortController.abort();
this.isFinished = true;
} }
/** /**
@@ -3239,9 +3240,12 @@ class StreamingProcessor {
console.warn(`Stream stats: ${timestamps.length} tokens, ${seconds.toFixed(2)} seconds, rate: ${Number(timestamps.length / seconds).toFixed(2)} TPS`); console.warn(`Stream stats: ${timestamps.length} tokens, ${seconds.toFixed(2)} seconds, rate: ${Number(timestamps.length / seconds).toFixed(2)} TPS`);
} }
catch (err) { catch (err) {
console.error(err); // in the case of a self-inflicted abort, we have already cleaned up
this.onErrorStreaming(); if (!this.isFinished) {
return; console.error(err);
this.onErrorStreaming();
}
return this.result;
} }
this.isFinished = true; this.isFinished = true;
@@ -4641,7 +4645,6 @@ export function stopGeneration() {
let stopped = false; let stopped = false;
if (streamingProcessor) { if (streamingProcessor) {
streamingProcessor.onStopStreaming(); streamingProcessor.onStopStreaming();
streamingProcessor = null;
stopped = true; stopped = true;
} }
if (abortController) { if (abortController) {