rixty
/
SillyTavern
mirror of https://github.com/SillyTavern/SillyTavern.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Move reroll button

This commit is contained in:
Cohee 2024-11-17 18:45:53 +02:00
parent 5e883e446a
commit de4246f7b7
3 changed files with 8 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
public/css/logprobs.css
View File

@ -80,12 +80,6 @@
background-color: rgba(255, 0, 50, 0.4);
}
#logprobsReroll {
float: right; /* Position the button to the right */
margin: 5px 0 5px 10px; /* Add spacing (top, right, bottom, left) */
clear: right; /* Ensure it starts on a new line */
}
.logprobs_candidate_list {
grid-row-start: 3;
grid-row-end: 4;

5
public/index.html
View File

@ -6631,8 +6631,11 @@
</div>
</div>
<div class="logprobs_panel_content inline-drawer-content flex-container flexFlowColumn">
<small>
<small class="flex-container alignItemsCenter justifySpaceBetween">
<b data-i18n="Select a token to see alternatives considered by the AI.">Select a token to see alternatives considered by the AI.</b>
<button id="logprobsReroll" class="menu_button" title="Reroll with the entire prefix" data-i18n="[title]Reroll with the entire prefix">
<span class="fa-solid fa-redo logprobs_reroll"></span>
</button>
</small>
<hr>
<div id="logprobs_generation_output"></div>

11
public/scripts/logprobs.js
View File

@ -17,6 +17,7 @@ import { t } from './i18n.js';
const TINTS = 4;
const MAX_MESSAGE_LOGPROBS = 100;
const REROLL_BUTTON = $('#logprobsReroll');
/**
* Tuple of a candidate token and its logarithm of probability of being chosen
@ -93,14 +94,10 @@ function renderAlternativeTokensView() {
const prefix = continueFrom || '';
const tokenSpans = [];
REROLL_BUTTON.toggle(!!prefix);
if (prefix) {
const rerollButton = $('<button id="logprobsReroll" class="menu_button">' +
' <span class="fa-solid fa-redo logprobs_reroll"></span>' +
'</button>');
rerollButton.attr('title', t`Reroll with the entire prefix`);
rerollButton.on('click', () => onPrefixClicked(prefix.length));
tokenSpans.push(rerollButton);
REROLL_BUTTON.off('click').on('click', () => onPrefixClicked(prefix.length));
let cumulativeOffset = 0;
const words = prefix.split(/\s+/);
@ -127,7 +124,6 @@ function renderAlternativeTokensView() {
});
}
messageLogprobs.forEach((tokenData, i) => {
const { token } = tokenData;
const span = $('<span></span>');
@ -539,6 +535,7 @@ function convertTokenIdLogprobsToText(input) {
}
export function initLogprobs() {
REROLL_BUTTON.hide();
const debouncedRender = debounce(renderAlternativeTokensView);
$('#logprobsViewerClose').on('click', onToggleLogprobsPanel);
$('#option_toggle_logprobs').on('click', onToggleLogprobsPanel);