make sampler nuke button work with zensliders

This commit is contained in:
RossAscends
2024-01-14 18:33:40 +09:00
parent 80b39bdac1
commit c6487fa9cb

View File

@ -504,45 +504,57 @@ jQuery(function () {
$('#samplerResetButton').off('click').on('click', function () { $('#samplerResetButton').off('click').on('click', function () {
const inputs = { const inputs = {
'temp_textgenerationwebui': '1', 'temp_textgenerationwebui': 1,
'top_k_textgenerationwebui': '0', 'top_k_textgenerationwebui': 0,
'top_p_textgenerationwebui': '1', 'top_p_textgenerationwebui': 1,
'min_p_textgenerationwebui': '0', 'min_p_textgenerationwebui': 0,
'rep_pen_textgenerationwebui': '1', 'rep_pen_textgenerationwebui': 1,
'rep_pen_range_textgenerationwebui': '0', 'rep_pen_range_textgenerationwebui': 0,
'dynatemp_textgenerationwebui': false, 'dynatemp_textgenerationwebui': false,
'seed_textgenerationwebui': '-1', 'seed_textgenerationwebui': 1,
'ban_eos_token_textgenerationwebui': false, 'ban_eos_token_textgenerationwebui': false,
'do_sample_textgenerationwebui': true, 'do_sample_textgenerationwebui': true,
'add_bos_token_textgenerationwebui': true, 'add_bos_token_textgenerationwebui': true,
'temperature_last_textgenerationwebui': true, 'temperature_last_textgenerationwebui': true,
'skip_special_tokens_textgenerationwebui': true, 'skip_special_tokens_textgenerationwebui': true,
'top_a_textgenerationwebui': '0', 'top_a_textgenerationwebui': 0,
'top_a_counter_textgenerationwebui': '0', 'top_a_counter_textgenerationwebui': 0,
'mirostat_mode_textgenerationwebui': '0', 'mirostat_mode_textgenerationwebui': 0,
'mirostat_tau_textgenerationwebui': '5', 'mirostat_tau_textgenerationwebui': 5,
'mirostat_eta_textgenerationwebui': '0.1', 'mirostat_eta_textgenerationwebui': 0.1,
'tfs_textgenerationwebui': '1', 'tfs_textgenerationwebui': 1,
'epsilon_cutoff_textgenerationwebui': '0', 'epsilon_cutoff_textgenerationwebui': 0,
'eta_cutoff_textgenerationwebui': '0', 'eta_cutoff_textgenerationwebui': 0,
'encoder_rep_pen_textgenerationwebui': '1', 'encoder_rep_pen_textgenerationwebui': 1,
'freq_pen_textgenerationwebui': '0', 'freq_pen_textgenerationwebui': 0,
'presence_pen_textgenerationwebui': '0', 'presence_pen_textgenerationwebui': 0,
'no_repeat_ngram_size_textgenerationwebui': '0', 'no_repeat_ngram_size_textgenerationwebui': 0,
'min_length_textgenerationwebui': '0', 'min_length_textgenerationwebui': 0,
'num_beams_textgenerationwebui': '1', 'num_beams_textgenerationwebui': 1,
'length_penalty_textgenerationwebui': '0', 'length_penalty_textgenerationwebui': 0,
'penalty_alpha_textgenerationwebui': '0', 'penalty_alpha_textgenerationwebui': 0,
'typical_p_textgenerationwebui': '1', // Added entry 'typical_p_textgenerationwebui': 1, // Added entry
'guidance_scale_textgenerationwebui': '1', 'guidance_scale_textgenerationwebui': 1,
}; };
for (const [id, value] of Object.entries(inputs)) { for (const [id, value] of Object.entries(inputs)) {
const inputElement = $(`#${id}`); const inputElement = $(`#${id}`);
if (inputElement.prop('type') === 'checkbox') { if (inputElement.prop('type') === 'checkbox') {
inputElement.prop('checked', value); inputElement.prop('checked', value);
} else if (inputElement.prop('type') === 'number') {
inputElement.val(value).trigger('input');
} else { } else {
inputElement.val(value).trigger('input'); inputElement.val(value).trigger('input');
if (power_user.enableZenSliders) {
let masterElementID = inputElement.prop('id');
console.log(masterElementID)
let zenSlider = $(`#${masterElementID}_zenslider`).slider();
zenSlider.slider('option', 'value', value);
zenSlider.slider('option', 'slide')
.call(zenSlider, null, {
handle: $('.ui-slider-handle', zenSlider), value: value,
});
}
} }
} }
}); });