diff --git a/public/index.html b/public/index.html index 34179f3fe..abf9ba3d0 100644 --- a/public/index.html +++ b/public/index.html @@ -1305,6 +1305,11 @@ +
+ Min P + + +
Top A @@ -5054,4 +5059,4 @@ - + \ No newline at end of file diff --git a/public/scripts/power-user.js b/public/scripts/power-user.js index c820ad41c..99bee5c09 100644 --- a/public/scripts/power-user.js +++ b/public/scripts/power-user.js @@ -496,7 +496,8 @@ async function switchZenSliders() { numSteps = 20 decimals = 1 } - if (sliderID == 'tfs_textgenerationwebui') { + if (sliderID == 'tfs_textgenerationwebui' || + sliderID == 'min_p_textgenerationwebui') { numSteps = 20 decimals = 2 } diff --git a/public/scripts/textgen-settings.js b/public/scripts/textgen-settings.js index adc2b90a4..a5f2b7f07 100644 --- a/public/scripts/textgen-settings.js +++ b/public/scripts/textgen-settings.js @@ -36,6 +36,7 @@ const textgenerationwebui_settings = { epsilon_cutoff: 0, eta_cutoff: 0, typical_p: 1, + min_p: 0, rep_pen: 1.2, rep_pen_range: 0, no_repeat_ngram_size: 0, @@ -91,6 +92,7 @@ const setting_names = [ "epsilon_cutoff", "eta_cutoff", "typical_p", + "min_p", "penalty_alpha", "num_beams", "length_penalty", @@ -122,7 +124,7 @@ const setting_names = [ //'prompt_log_probs_aphrodite' ]; -function selectPreset(name) { +async function selectPreset(name) { const preset = textgenerationwebui_presets[textgenerationwebui_preset_names.indexOf(name)]; if (!preset) { @@ -351,6 +353,14 @@ function setSettingByName(i, value, trigger) { const val = parseFloat(value); $(`#${i}_textgenerationwebui`).val(val); $(`#${i}_counter_textgenerationwebui`).val(val); + if (power_user.enableZenSliders) { + let zenSlider = $(`#${i}_textgenerationwebui_zenslider`).slider() + zenSlider.slider('option', 'value', val) + zenSlider.slider('option', 'slide') + .call(zenSlider, null, { + handle: $('.ui-slider-handle', zenSlider), value: val + }); + } } if (trigger) { @@ -448,6 +458,7 @@ export function getTextGenGenerationData(finalPrompt, this_amount_gen, isImperso 'temperature': textgenerationwebui_settings.temp, 'top_p': textgenerationwebui_settings.top_p, 'typical_p': textgenerationwebui_settings.typical_p, + 'min_p': textgenerationwebui_settings.min_p, 'repetition_penalty': textgenerationwebui_settings.rep_pen, 'repetition_penalty_range': textgenerationwebui_settings.rep_pen_range, 'encoder_repetition_penalty': textgenerationwebui_settings.encoder_rep_pen,