Add Min P for Novel

8efe0b6437a47e75b423030b8ec854e2f3121984

Cohee <18619528+Cohee1207@users.noreply.github.com>

Parents: 6d32cd6bf095ffc9a3a9b31731d7cb97318a67f8
22 files changed, +64 -19Showing whitespace changes
default/content/presets/novel/Asper-Kayra.json+2 -1
@@ -16,5 +16,6 @@
1616 "prefix": "vanilla",
1717 "cfg_scale": 1,
1818 "phrase_rep_pen": "medium",
1919 "max_context": 7800,
20+ "min_p": 0
2021}
default/content/presets/novel/Blended-Coffee-Kayra.json+2 -1
@@ -15,5 +15,6 @@
1515 "prefix": "vanilla",
1616 "phrase_rep_pen": "medium",
1717 "cfg_scale": 1.55,
1818 "max_context": 7800,
19+ "min_p": 0
1920}
default/content/presets/novel/Blook-Kayra.json+2 -1
@@ -16,5 +16,6 @@
1616 "prefix": "vanilla",
1717 "phrase_rep_pen": "very_aggressive",
1818 "cfg_scale": 1.3,
1919 "max_context": 7800,
20+ "min_p": 0
2021}
default/content/presets/novel/Carefree-Kayra.json+2 -1
@@ -17,5 +17,6 @@
1717 "prefix": "vanilla",
1818 "cfg_scale": 1,
1919 "phrase_rep_pen": "aggressive",
2020 "max_context": 7800,
21+ "min_p": 0
2122}
default/content/presets/novel/CosmicCube-Kayra.json+2 -1
@@ -16,5 +16,6 @@
1616 "prefix": "vanilla",
1717 "phrase_rep_pen": "off",
1818 "cfg_scale": 1.48,
1919 "max_context": 7800,
20+ "min_p": 0
2021}
default/content/presets/novel/Edgewise-Clio.json+2 -1
@@ -17,5 +17,6 @@
1717 "prefix": "vanilla",
1818 "cfg_scale": 1,
1919 "phrase_rep_pen": "very_light",
2020 "max_context": 7800,
21+ "min_p": 0
2122}
default/content/presets/novel/Fresh-Coffee-Clio.json+2 -1
@@ -18,5 +18,6 @@
1818 "prefix": "vanilla",
1919 "cfg_scale": 1,
2020 "phrase_rep_pen": "very_light",
2121 "max_context": 7800,
22+ "min_p": 0
2223}
default/content/presets/novel/Fresh-Coffee-Kayra.json+2 -1
@@ -16,5 +16,6 @@
1616 "prefix": "vanilla",
1717 "cfg_scale": 1,
1818 "phrase_rep_pen": "off",
1919 "max_context": 7800,
20+ "min_p": 0
2021}
default/content/presets/novel/Green-Active-Writer-Kayra.json+2 -1
@@ -16,5 +16,6 @@
1616 "prefix": "vanilla",
1717 "phrase_rep_pen": "very_aggressive",
1818 "cfg_scale": 1.4,
1919 "max_context": 7800,
20+ "min_p": 0
2021}
default/content/presets/novel/Keelback-Clio.json+2 -1
@@ -18,5 +18,6 @@
1818 "prefix": "vanilla",
1919 "cfg_scale": 1,
2020 "phrase_rep_pen": "very_light",
2121 "max_context": 7800,
22+ "min_p": 0
2223}
default/content/presets/novel/Long-Press-Clio.json+2 -1
@@ -18,5 +18,6 @@
1818 "prefix": "vanilla",
1919 "cfg_scale": 1,
2020 "phrase_rep_pen": "very_light",
2121 "max_context": 7800,
22+ "min_p": 0
2223}
default/content/presets/novel/Pilotfish-Kayra.json+2 -1
@@ -18,5 +18,6 @@
1818 "prefix": "vanilla",
1919 "phrase_rep_pen": "medium",
2020 "cfg_scale": 1.35,
2121 "max_context": 7800,
22+ "min_p": 0
2223}
default/content/presets/novel/Pro_Writer-Kayra.json+2 -1
@@ -16,5 +16,6 @@
1616 "prefix": "vanilla",
1717 "phrase_rep_pen": "medium",
1818 "cfg_scale": 1.0,
1919 "max_context": 7800,
20+ "min_p": 0
2021}
default/content/presets/novel/Stelenes-Kayra.json+2 -1
@@ -13,5 +13,6 @@
1313 "return_full_text": false,
1414 "prefix": "vanilla",
1515 "phrase_rep_pen": "medium",
1616 "max_context": 7800,
17+ "min_p": 0
1718}
default/content/presets/novel/Talker-Chat-Clio.json+2 -1
@@ -18,5 +18,6 @@
1818 "prefix": "vanilla",
1919 "cfg_scale": 1,
2020 "phrase_rep_pen": "very_light",
2121 "max_context": 7800,
22+ "min_p": 0
2223}
default/content/presets/novel/Tea_Time-Kayra.json+2 -1
@@ -15,5 +15,6 @@
1515 "prefix": "vanilla",
1616 "phrase_rep_pen": "aggressive",
1717 "cfg_scale": 1.0,
1818 "max_context": 7800,
19+ "min_p": 0
1920}
default/content/presets/novel/Tesseract-Kayra.json+2 -1
@@ -14,5 +14,6 @@
1414 "prefix": "vanilla",
1515 "phrase_rep_pen": "aggressive",
1616 "cfg_scale": 1.3,
1717 "max_context": 7800,
18+ "min_p": 0
1819}
default/content/presets/novel/Vingt-Un-Clio.json+2 -1
@@ -18,5 +18,6 @@
1818 "prefix": "vanilla",
1919 "cfg_scale": 1,
2020 "phrase_rep_pen": "very_light",
2121 "max_context": 7800,
22+ "min_p": 0
2223}
default/content/presets/novel/Writers-Daemon-Kayra.json+2 -1
@@ -17,5 +17,6 @@
1717 "return_full_text": false,
1818 "prefix": "vanilla",
1919 "phrase_rep_pen": "very_aggressive",
2020 "max_context": 7800,
21+ "min_p": 0
2122}
public/index.html+13 -0
@@ -324,6 +324,19 @@
324324 </div>
325325 </div>
326326 <div class="range-block">
327+ <div class="range-block-title" data-i18n="Min P">
328+ Min P
329+ </div>
330+ <div class="range-block-range-and-counter">
331+ <div class="range-block-range">
332+ <input type="range" id="min_p_novel" name="volume" min="0" max="1" step="0.001">
333+ </div>
334+ <div class="range-block-counter">
335+ <input type="number" min="0" max="1" step="0.001" data-for="min_p_novel" id="min_p_counter_novel">
336+ </div>
337+ </div>
338+ </div>
339+ <div class="range-block">
327340 <div class="range-block-title" data-i18n="TFS">
328341 TFS
329342 </div>
public/scripts/nai-settings.js+12 -0
@@ -38,6 +38,7 @@ export const nai_settings = {
3838 top_p: 0.75,
3939 top_a: 0.08,
4040 typical_p: 0.975,
41+ min_p: 0,
4142 min_length: 1,
4243 model_novel: 'clio-v1',
4344 preset_settings_novel: 'Talker-Chat-Clio',
@@ -151,6 +152,7 @@ export function loadNovelPreset(preset) {
151152 nai_settings.order = preset.order || default_order;
152153 nai_settings.logit_bias = preset.logit_bias || [];
153154 nai_settings.preamble = preset.preamble || default_preamble;
155+ nai_settings.min_p = preset.min_p || 0;
154156 loadNovelSettingsUi(nai_settings);
155157}
156158
@@ -188,6 +190,7 @@ export function loadNovelSettings(settings) {
188190 nai_settings.banned_tokens = settings.banned_tokens || '';
189191 nai_settings.order = settings.order || default_order;
190192 nai_settings.logit_bias = settings.logit_bias || [];
193+ nai_settings.min_p = settings.min_p || 0;
191194 loadNovelSettingsUi(nai_settings);
192195}
193196
@@ -227,6 +230,8 @@ function loadNovelSettingsUi(ui_settings) {
227230 $('#nai_prefix').val(ui_settings.prefix || 'vanilla');
228231 $('#nai_cfg_uc').val(ui_settings.cfg_uc || '');
229232 $('#nai_banned_tokens').val(ui_settings.banned_tokens || '');
233+ $('#min_p_novel').val(ui_settings.min_p);
234+ $('#min_p_counter_novel').val(Number(ui_settings.min_p).toFixed(3));
230235
231236 $('#streaming_novel').prop('checked', ui_settings.streaming_novel);
232237 sortItemsByOrder(ui_settings.order);
@@ -336,6 +341,12 @@ const sliders = [
336341 format: (val) => val,
337342 setValue: (val) => { nai_settings.banned_tokens = val; },
338343 },
344+ {
345+ sliderId: '#min_p_novel',
346+ counterId: '#min_p_counter_novel',
347+ format: (val) => Number(val).toFixed(3),
348+ setValue: (val) => { nai_settings.min_p = Number(val); },
349+ },
339350];
340351
341352function getBadWordIds(banned_tokens, tokenizerType) {
@@ -469,6 +480,7 @@ export function getNovelGenerationData(finalPrompt, settings, maxLength, isImper
469480 'top_a': Number(nai_settings.top_a),
470481 'top_p': Number(nai_settings.top_p),
471482 'top_k': Number(nai_settings.top_k),
483+ 'min_p': Number(nai_settings.min_p),
472484 'typical_p': Number(nai_settings.typical_p),
473485 'mirostat_lr': Number(nai_settings.mirostat_lr),
474486 'mirostat_tau': Number(nai_settings.mirostat_tau),
src/endpoints/novelai.js+1 -0
@@ -175,6 +175,7 @@ router.post('/generate', jsonParser, async function (req, res) {
175175 'prefix': req.body.prefix,
176176 'order': req.body.order,
177177 'num_logprobs': req.body.num_logprobs,
178+ 'min_p': req.body.min_p,
178179 },
179180 };
180181