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

add killswitch for banned strings

Browse Source
This commit is contained in:
RossAscends
2025-02-02 12:00:01 +09:00
parent 890c0c4723
commit c5dad20fc4
3 changed files with 56 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
public/scripts/textgen-settings.js
View File

@ -182,6 +182,7 @@ const settings = {
grammar_string: '',
json_schema: {},
banned_tokens: '',
sendBannedTokens: true,
sampler_priority: OOBA_DEFAULT_ORDER,
samplers: LLAMACPP_DEFAULT_ORDER,
samplers_priorities: APHRODITE_DEFAULT_ORDER,
@ -274,6 +275,7 @@ export const setting_names = [
'grammar_string',
'json_schema',
'banned_tokens',
'sendBannedTokens',
'ignore_eos_token',
'spaces_between_special_tokens',
'speculative_ngram',
@ -452,6 +454,22 @@ function getCustomTokenBans() {
};
}
async function enableBannedStringsKillSwitch() {
$('#bannedStringsKillSwitch_textgenerationwebui').prop('checked', true);
$('#bannedStringsKillSwitch_label').find('.menu_button').addClass('toggleEnabled').prop('title', 'Banned tokens/strings are being sent into the request.');
settings.sendBannedTokens = true;
saveSettingsDebounced();
return '';
}
async function disableBannedStringsKillSwitch() {
$('#bannedStringsKillSwitch_textgenerationwebui').prop('checked', false);
$('#bannedStringsKillSwitch_label').find('.menu_button').removeClass('toggleEnabled').prop('title', 'Banned tokens/strings are NOT being sent into the request.');
settings.sendBannedTokens = false;
saveSettingsDebounced();
return '';
}
/**
* Calculates logit bias object from the logit bias list.
* @returns {object} Logit bias object
@ -594,6 +612,16 @@ function sortAphroditeItemsByOrder(orderArray) {
}
jQuery(function () {
$('#bannedStringsKillSwitch_textgenerationwebui').on('change', function () {
const checked = $(this).prop('checked');
if (checked) {
enableBannedStringsKillSwitch();
} else {
disableBannedStringsKillSwitch();
}
});
$('#koboldcpp_order').sortable({
delay: getSortableDelay(),
stop: function () {
@ -927,6 +955,11 @@ function setSettingByName(setting, value, trigger) {
return;
}
if ('sendBannedTokens' === setting) {
$('#bannedStringsKillSwitch_textgenerationwebui').prop('checked', value).trigger('change');
return;
}
const isCheckbox = $(`#${setting}_textgenerationwebui`).attr('type') == 'checkbox';
const isText = $(`#${setting}_textgenerationwebui`).attr('type') == 'text' || $(`#${setting}_textgenerationwebui`).is('textarea');
if (isCheckbox) {
@ -1461,5 +1494,9 @@ export function getTextGenGenerationData(finalPrompt, maxTokens, isImpersonate,
}
}
//last check before sending sampler params, strip banned strings if toggled off.
const sendBannedStrings = $('#bannedStringsKillSwitch_textgenerationwebui').prop('checked');
if (!sendBannedStrings) { delete params.banned_strings; }
return params;
}