Also clear prompt cache on streaming stop

This commit is contained in:
Cohee 2024-03-28 20:22:14 +02:00
parent b080f8af48
commit 7c99d87238
1 changed files with 3 additions and 8 deletions

View File

@ -2733,9 +2733,7 @@ class StreamingProcessor {
const continueMsg = this.type === 'continue' ? this.messageAlreadyGenerated : undefined; const continueMsg = this.type === 'continue' ? this.messageAlreadyGenerated : undefined;
saveLogprobsForActiveMessage(this.messageLogprobs.filter(Boolean), continueMsg); saveLogprobsForActiveMessage(this.messageLogprobs.filter(Boolean), continueMsg);
await saveChatConditional(); await saveChatConditional();
activateSendButtons(); unblockGeneration();
showSwipeButtons();
setGenerationProgress(0);
generatedPromptCache = ''; generatedPromptCache = '';
//console.log("Generated text size:", text.length, text) //console.log("Generated text size:", text.length, text)
@ -2778,11 +2776,8 @@ class StreamingProcessor {
this.isStopped = true; this.isStopped = true;
this.hideMessageButtons(this.messageId); this.hideMessageButtons(this.messageId);
$('#send_textarea').removeAttr('disabled'); generatedPromptCache = '';
is_send_press = false; unblockGeneration();
activateSendButtons();
setGenerationProgress(0);
showSwipeButtons();
} }
setFirstSwipe(messageId) { setFirstSwipe(messageId) {