diff --git a/public/img/infermaticai.svg b/public/img/infermaticai.svg new file mode 100644 index 000000000..7f88c643c --- /dev/null +++ b/public/img/infermaticai.svg @@ -0,0 +1,40 @@ + + + + + + diff --git a/public/scripts/textgen-settings.js b/public/scripts/textgen-settings.js index c2666f41d..a35763905 100644 --- a/public/scripts/textgen-settings.js +++ b/public/scripts/textgen-settings.js @@ -578,7 +578,7 @@ jQuery(function () { const type = String($(this).val()); settings.type = type; - if (settings.type === textgen_types.VLLM || settings.type === textgen_types.APHRODITE) { + if ([VLLM, APHRODITE, INFERMATICAI].includes(settings.type)) { $('#mirostat_mode_textgenerationwebui').attr('step', 2); //Aphro disallows mode 1 $('#do_sample_textgenerationwebui').prop('checked', true); //Aphro should always do sample; 'otherwise set temp to 0 to mimic no sample' $('#ban_eos_token_textgenerationwebui').prop('checked', false); //Aphro should not ban EOS, just ignore it; 'add token '2' to ban list do to this' @@ -619,7 +619,7 @@ jQuery(function () { $('#samplerResetButton').off('click').on('click', function () { const inputs = { 'temp_textgenerationwebui': 1, - 'top_k_textgenerationwebui': 0, + 'top_k_textgenerationwebui': [INFERMATICAI, APHRODITE, VLLM].includes(settings.type) ? -1 : 0, 'top_p_textgenerationwebui': 1, 'min_p_textgenerationwebui': 0, 'rep_pen_textgenerationwebui': 1,