Add new NAI samplers and update presets
This commit is contained in:
parent
93123062af
commit
a0573413b2
|
@ -1,19 +1,20 @@
|
||||||
{
|
{
|
||||||
"order": [5, 0, 1, 3],
|
"order": [5, 0, 1, 3, 7],
|
||||||
"temperature": 1.23,
|
"temperature": 1.35,
|
||||||
"max_length": 300,
|
"max_length": 300,
|
||||||
"min_length": 1,
|
"min_length": 1,
|
||||||
"top_k": 200,
|
"top_k": 225,
|
||||||
"typical_p": 0.966,
|
"top_g": 8,
|
||||||
"tail_free_sampling": 0.982,
|
"typical_p": 0.975,
|
||||||
"repetition_penalty": 1.74,
|
"tail_free_sampling": 0.984,
|
||||||
"repetition_penalty_range": 4000,
|
"repetition_penalty": 1.7,
|
||||||
|
"repetition_penalty_range": 3200,
|
||||||
"repetition_penalty_frequency": 0,
|
"repetition_penalty_frequency": 0,
|
||||||
"repetition_penalty_presence": 0.02,
|
"repetition_penalty_presence": 0.02,
|
||||||
"use_cache": false,
|
"use_cache": false,
|
||||||
"return_full_text": false,
|
"return_full_text": false,
|
||||||
"prefix": "vanilla",
|
"prefix": "vanilla",
|
||||||
"cfg_scale": 1,
|
"cfg_scale": 1,
|
||||||
"phrase_rep_pen": "aggressive",
|
"phrase_rep_pen": "medium",
|
||||||
"max_context": 7800
|
"max_context": 7800
|
||||||
}
|
}
|
||||||
|
|
|
@ -3,7 +3,7 @@
|
||||||
"temperature": 1.35,
|
"temperature": 1.35,
|
||||||
"max_length": 300,
|
"max_length": 300,
|
||||||
"min_length": 1,
|
"min_length": 1,
|
||||||
"top_k": 12,
|
"top_k": 15,
|
||||||
"top_p": 0.85,
|
"top_p": 0.85,
|
||||||
"top_a": 0.1,
|
"top_a": 0.1,
|
||||||
"tail_free_sampling": 0.915,
|
"tail_free_sampling": 0.915,
|
||||||
|
|
|
@ -1,19 +1,20 @@
|
||||||
{
|
{
|
||||||
"order": [6, 1, 0, 5, 3],
|
"order": [6, 0, 8, 5, 3],
|
||||||
"temperature": 1.25,
|
"temperature": 1.5,
|
||||||
"max_length": 300,
|
"max_length": 300,
|
||||||
"min_length": 1,
|
"min_length": 1,
|
||||||
"top_k": 70,
|
"typical_p": 0.95,
|
||||||
"typical_p": 0.9,
|
"tail_free_sampling": 0.95,
|
||||||
"tail_free_sampling": 0.925,
|
"mirostat_lr": 0.2,
|
||||||
"repetition_penalty": 2,
|
"mirostat_tau": 5.5,
|
||||||
|
"repetition_penalty": 1,
|
||||||
"repetition_penalty_range": 1632,
|
"repetition_penalty_range": 1632,
|
||||||
"repetition_penalty_frequency": 0,
|
"repetition_penalty_frequency": 0,
|
||||||
"repetition_penalty_presence": 0,
|
"repetition_penalty_presence": 0,
|
||||||
"use_cache": false,
|
"use_cache": false,
|
||||||
"return_full_text": false,
|
"return_full_text": false,
|
||||||
"prefix": "vanilla",
|
"prefix": "vanilla",
|
||||||
"phrase_rep_pen": "aggressive",
|
"phrase_rep_pen": "very_aggressive",
|
||||||
"cfg_scale": 1.825,
|
"cfg_scale": 1.4,
|
||||||
"max_context": 7800
|
"max_context": 7800
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,12 +1,13 @@
|
||||||
{
|
{
|
||||||
"order": [3, 4, 0],
|
"order": [3, 4, 5, 0],
|
||||||
"temperature": 1.19,
|
"temperature": 1.06,
|
||||||
"max_length": 300,
|
"max_length": 300,
|
||||||
"min_length": 1,
|
"min_length": 1,
|
||||||
"top_a": 0.116,
|
"top_a": 0.146,
|
||||||
"tail_free_sampling": 0.958,
|
"typical_p": 0.976,
|
||||||
"repetition_penalty": 1.64,
|
"tail_free_sampling": 0.969,
|
||||||
"repetition_penalty_slope": 2.12,
|
"repetition_penalty": 1.86,
|
||||||
|
"repetition_penalty_slope": 2.33,
|
||||||
"repetition_penalty_frequency": 0,
|
"repetition_penalty_frequency": 0,
|
||||||
"repetition_penalty_presence": 0,
|
"repetition_penalty_presence": 0,
|
||||||
"repetition_penalty_range": 2048,
|
"repetition_penalty_range": 2048,
|
||||||
|
|
|
@ -1,12 +1,14 @@
|
||||||
{
|
{
|
||||||
"order": [6, 1, 0, 5, 3, 2],
|
"order": [8, 0, 5, 3, 2, 4],
|
||||||
"temperature": 1.5,
|
"temperature": 1.5,
|
||||||
"max_length": 300,
|
"max_length": 300,
|
||||||
"min_length": 1,
|
"min_length": 1,
|
||||||
"top_k": 70,
|
"top_a": 0.02,
|
||||||
"top_p": 0.95,
|
"top_p": 0.95,
|
||||||
"typical_p": 0.95,
|
"typical_p": 0.95,
|
||||||
"tail_free_sampling": 0.95,
|
"tail_free_sampling": 0.95,
|
||||||
|
"mirostat_lr": 0.2,
|
||||||
|
"mirostat_tau": 5,
|
||||||
"repetition_penalty": 1.6,
|
"repetition_penalty": 1.6,
|
||||||
"repetition_penalty_range": 2016,
|
"repetition_penalty_range": 2016,
|
||||||
"repetition_penalty_frequency": 0,
|
"repetition_penalty_frequency": 0,
|
||||||
|
|
|
@ -896,6 +896,51 @@
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
<div class="range-block">
|
||||||
|
<div class="range-block-title" data-i18n="Top G">
|
||||||
|
Top G
|
||||||
|
</div>
|
||||||
|
<div class="range-block-range-and-counter">
|
||||||
|
<div class="range-block-range">
|
||||||
|
<input type="range" id="top_g_novel" name="volume" min="0" max="20" step="1">
|
||||||
|
</div>
|
||||||
|
<div class="range-block-counter">
|
||||||
|
<div contenteditable="true" data-for="top_g_novel" id="top_g_counter_novel">
|
||||||
|
select
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="range-block">
|
||||||
|
<div class="range-block-title" data-i18n="Mirostat Tau">
|
||||||
|
Mirostat Tau
|
||||||
|
</div>
|
||||||
|
<div class="range-block-range-and-counter">
|
||||||
|
<div class="range-block-range">
|
||||||
|
<input type="range" id="mirostat_tau_novel" name="volume" min="0" max="6" step="0.01">
|
||||||
|
</div>
|
||||||
|
<div class="range-block-counter">
|
||||||
|
<div contenteditable="true" data-for="mirostat_tau_novel" id="mirostat_tau_counter_novel">
|
||||||
|
select
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div class="range-block">
|
||||||
|
<div class="range-block-title" data-i18n="Mirostat LR">
|
||||||
|
Mirostat LR
|
||||||
|
</div>
|
||||||
|
<div class="range-block-range-and-counter">
|
||||||
|
<div class="range-block-range">
|
||||||
|
<input type="range" id="mirostat_lr_novel" name="volume" min="0" max="1" step="0.01">
|
||||||
|
</div>
|
||||||
|
<div class="range-block-counter">
|
||||||
|
<div contenteditable="true" data-for="mirostat_lr_novel" id="mirostat_lr_counter_novel">
|
||||||
|
select
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
<div class="range-block">
|
<div class="range-block">
|
||||||
<div class="range-block-title" data-i18n="Typical P">
|
<div class="range-block-title" data-i18n="Typical P">
|
||||||
Typical P
|
Typical P
|
||||||
|
|
|
@ -80,6 +80,9 @@ function loadNovelPreset(preset) {
|
||||||
nai_settings.min_length = preset.min_length;
|
nai_settings.min_length = preset.min_length;
|
||||||
nai_settings.cfg_scale = preset.cfg_scale;
|
nai_settings.cfg_scale = preset.cfg_scale;
|
||||||
nai_settings.phrase_rep_pen = preset.phrase_rep_pen;
|
nai_settings.phrase_rep_pen = preset.phrase_rep_pen;
|
||||||
|
nai_settings.top_g = preset.top_g;
|
||||||
|
nai_settings.mirostat_lr = preset.mirostat_lr;
|
||||||
|
nai_settings.mirostat_tau = preset.mirostat_tau;
|
||||||
loadNovelSettingsUi(nai_settings);
|
loadNovelSettingsUi(nai_settings);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -105,6 +108,9 @@ function loadNovelSettings(settings) {
|
||||||
nai_settings.min_length = settings.min_length;
|
nai_settings.min_length = settings.min_length;
|
||||||
nai_settings.phrase_rep_pen = settings.phrase_rep_pen;
|
nai_settings.phrase_rep_pen = settings.phrase_rep_pen;
|
||||||
nai_settings.cfg_scale = settings.cfg_scale;
|
nai_settings.cfg_scale = settings.cfg_scale;
|
||||||
|
nai_settings.top_g = settings.top_g;
|
||||||
|
nai_settings.mirostat_lr = settings.mirostat_lr;
|
||||||
|
nai_settings.mirostat_tau = settings.mirostat_tau;
|
||||||
nai_settings.streaming_novel = !!settings.streaming_novel;
|
nai_settings.streaming_novel = !!settings.streaming_novel;
|
||||||
loadNovelSettingsUi(nai_settings);
|
loadNovelSettingsUi(nai_settings);
|
||||||
}
|
}
|
||||||
|
@ -169,6 +175,12 @@ function loadNovelSettingsUi(ui_settings) {
|
||||||
$("#cfg_scale_counter_novel").text(Number(ui_settings.cfg_scale).toFixed(2));
|
$("#cfg_scale_counter_novel").text(Number(ui_settings.cfg_scale).toFixed(2));
|
||||||
$("#phrase_rep_pen_novel").val(getPhraseRepPenCounter(ui_settings.phrase_rep_pen));
|
$("#phrase_rep_pen_novel").val(getPhraseRepPenCounter(ui_settings.phrase_rep_pen));
|
||||||
$("#phrase_rep_pen_counter_novel").text(getPhraseRepPenCounter(ui_settings.phrase_rep_pen));
|
$("#phrase_rep_pen_counter_novel").text(getPhraseRepPenCounter(ui_settings.phrase_rep_pen));
|
||||||
|
$("#top_g_novel").val(ui_settings.top_g);
|
||||||
|
$("#top_g_counter_novel").text(Number(ui_settings.top_g).toFixed(0));
|
||||||
|
$("#mirostat_lr_novel").val(ui_settings.mirostat_lr);
|
||||||
|
$("#mirostat_lr_counter_novel").text(Number(ui_settings.mirostat_lr).toFixed(2));
|
||||||
|
$("#mirostat_tau_novel").val(ui_settings.mirostat_tau);
|
||||||
|
$("#mirostat_tau_counter_novel").text(Number(ui_settings.mirostat_tau).toFixed(2));
|
||||||
$("#min_length_novel").val(ui_settings.min_length);
|
$("#min_length_novel").val(ui_settings.min_length);
|
||||||
$("#min_length_counter_novel").text(Number(ui_settings.min_length).toFixed(0));
|
$("#min_length_counter_novel").text(Number(ui_settings.min_length).toFixed(0));
|
||||||
$('#nai_preamble_textarea').val(ui_settings.nai_preamble);
|
$('#nai_preamble_textarea').val(ui_settings.nai_preamble);
|
||||||
|
@ -243,6 +255,24 @@ const sliders = [
|
||||||
format: (val) => Number(val).toFixed(2),
|
format: (val) => Number(val).toFixed(2),
|
||||||
setValue: (val) => { nai_settings.typical_p = Number(val).toFixed(2); },
|
setValue: (val) => { nai_settings.typical_p = Number(val).toFixed(2); },
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
sliderId: "#top_g_novel",
|
||||||
|
counterId: "#top_g_counter_novel",
|
||||||
|
format: (val) => Number(val).toFixed(0),
|
||||||
|
setValue: (val) => { nai_settings.top_g = Number(val).toFixed(0); },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
sliderId: "#mirostat_tau_novel",
|
||||||
|
counterId: "#mirostat_tau_counter_novel",
|
||||||
|
format: (val) => Number(val).toFixed(2),
|
||||||
|
setValue: (val) => { nai_settings.mirostat_tau = Number(val).toFixed(2); },
|
||||||
|
},
|
||||||
|
{
|
||||||
|
sliderId: "#mirostat_lr_novel",
|
||||||
|
counterId: "#mirostat_lr_counter_novel",
|
||||||
|
format: (val) => Number(val).toFixed(2),
|
||||||
|
setValue: (val) => { nai_settings.mirostat_lr = Number(val).toFixed(2); },
|
||||||
|
},
|
||||||
{
|
{
|
||||||
sliderId: "#cfg_scale_novel",
|
sliderId: "#cfg_scale_novel",
|
||||||
counterId: "#cfg_scale_counter_novel",
|
counterId: "#cfg_scale_counter_novel",
|
||||||
|
@ -298,6 +328,9 @@ export function getNovelGenerationData(finalPromt, this_settings, this_amount_ge
|
||||||
"top_p": parseFloat(nai_settings.top_p),
|
"top_p": parseFloat(nai_settings.top_p),
|
||||||
"top_k": parseInt(nai_settings.top_k),
|
"top_k": parseInt(nai_settings.top_k),
|
||||||
"typical_p": parseFloat(nai_settings.typical_p),
|
"typical_p": parseFloat(nai_settings.typical_p),
|
||||||
|
"top_g": parseFloat(nai_settings.top_g),
|
||||||
|
"mirostat_lr": parseFloat(nai_settings.mirostat_lr),
|
||||||
|
"mirostat_tau": parseFloat(nai_settings.mirostat_tau),
|
||||||
"cfg_scale": parseFloat(nai_settings.cfg_scale),
|
"cfg_scale": parseFloat(nai_settings.cfg_scale),
|
||||||
"cfg_uc": "",
|
"cfg_uc": "",
|
||||||
"phrase_rep_pen": nai_settings.phrase_rep_pen,
|
"phrase_rep_pen": nai_settings.phrase_rep_pen,
|
||||||
|
|
|
@ -1872,6 +1872,9 @@ app.post("/generate_novelai", jsonParser, async function (request, response_gene
|
||||||
"top_p": request.body.top_p,
|
"top_p": request.body.top_p,
|
||||||
"top_k": request.body.top_k,
|
"top_k": request.body.top_k,
|
||||||
"typical_p": request.body.typical_p,
|
"typical_p": request.body.typical_p,
|
||||||
|
"top_g": request.body.top_g,
|
||||||
|
"mirostat_lr": request.body.mirostat_lr,
|
||||||
|
"mirostat_tau": request.body.mirostat_tau,
|
||||||
"cfg_scale": request.body.cfg_scale,
|
"cfg_scale": request.body.cfg_scale,
|
||||||
"cfg_uc": request.body.cfg_uc,
|
"cfg_uc": request.body.cfg_uc,
|
||||||
"phrase_rep_pen": request.body.phrase_rep_pen,
|
"phrase_rep_pen": request.body.phrase_rep_pen,
|
||||||
|
|
Loading…
Reference in New Issue