TFS
diff --git a/public/scripts/nai-settings.js b/public/scripts/nai-settings.js
index 6b59033a8..6ac80c2dd 100644
--- a/public/scripts/nai-settings.js
+++ b/public/scripts/nai-settings.js
@@ -38,6 +38,7 @@ export const nai_settings = {
top_p: 0.75,
top_a: 0.08,
typical_p: 0.975,
+ min_p: 0,
min_length: 1,
model_novel: 'clio-v1',
preset_settings_novel: 'Talker-Chat-Clio',
@@ -151,6 +152,7 @@ export function loadNovelPreset(preset) {
nai_settings.order = preset.order || default_order;
nai_settings.logit_bias = preset.logit_bias || [];
nai_settings.preamble = preset.preamble || default_preamble;
+ nai_settings.min_p = preset.min_p || 0;
loadNovelSettingsUi(nai_settings);
}
@@ -188,6 +190,7 @@ export function loadNovelSettings(settings) {
nai_settings.banned_tokens = settings.banned_tokens || '';
nai_settings.order = settings.order || default_order;
nai_settings.logit_bias = settings.logit_bias || [];
+ nai_settings.min_p = settings.min_p || 0;
loadNovelSettingsUi(nai_settings);
}
@@ -227,6 +230,8 @@ function loadNovelSettingsUi(ui_settings) {
$('#nai_prefix').val(ui_settings.prefix || 'vanilla');
$('#nai_cfg_uc').val(ui_settings.cfg_uc || '');
$('#nai_banned_tokens').val(ui_settings.banned_tokens || '');
+ $('#min_p_novel').val(ui_settings.min_p);
+ $('#min_p_counter_novel').val(Number(ui_settings.min_p).toFixed(3));
$('#streaming_novel').prop('checked', ui_settings.streaming_novel);
sortItemsByOrder(ui_settings.order);
@@ -336,6 +341,12 @@ const sliders = [
format: (val) => val,
setValue: (val) => { nai_settings.banned_tokens = val; },
},
+ {
+ sliderId: '#min_p_novel',
+ counterId: '#min_p_counter_novel',
+ format: (val) => Number(val).toFixed(3),
+ setValue: (val) => { nai_settings.min_p = Number(val); },
+ },
];
function getBadWordIds(banned_tokens, tokenizerType) {
@@ -469,6 +480,7 @@ export function getNovelGenerationData(finalPrompt, settings, maxLength, isImper
'top_a': Number(nai_settings.top_a),
'top_p': Number(nai_settings.top_p),
'top_k': Number(nai_settings.top_k),
+ 'min_p': Number(nai_settings.min_p),
'typical_p': Number(nai_settings.typical_p),
'mirostat_lr': Number(nai_settings.mirostat_lr),
'mirostat_tau': Number(nai_settings.mirostat_tau),
diff --git a/src/endpoints/novelai.js b/src/endpoints/novelai.js
index d21602cc2..14aba9b84 100644
--- a/src/endpoints/novelai.js
+++ b/src/endpoints/novelai.js
@@ -175,6 +175,7 @@ router.post('/generate', jsonParser, async function (req, res) {
'prefix': req.body.prefix,
'order': req.body.order,
'num_logprobs': req.body.num_logprobs,
+ 'min_p': req.body.min_p,
},
};