Merge pull request #2742 from SillyTavern/xtc-ooba Enable XTC for TextGen WebUI

c47e19866420b9d67239312d47beea072850800c

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

Signed

public/index.html+8 -1 Viewed

                                         </div>
                                     </div>
                                     <div data-tg-type="koboldcpp, aphrodite, tabby, ooba" id="xtc_block" class="wide100p">
                                         <h4 class="wide100p textAlignCenter">
                                             <label data-i18n="Exclude Top Choices (XTC)">Exclude Top Choices (XTC)</label>
                                             <a href="https://github.com/oobabooga/text-generation-webui/pull/6335" target="_blank">

                                             Ooba only. Determines the order of samplers.
                                         </div>
                                         <div id="sampler_priority_container" class="prompt_order">
 +                                            <div data-name="repetition_penalty" draggable="true"><span>Repetition Penalty</span><small></small></div>
 +                                            <div data-name="presence_penalty" draggable="true"><span>Presence Penalty</span><small></small></div>
 +                                            <div data-name="frequency_penalty" draggable="true"><span>Frequency Penalty</span><small></small></div>
 +                                            <div data-name="dry" draggable="true"><span>DRY</span><small></small></div>
                                             <div data-name="temperature" draggable="true"><span>Temperature</span><small></small></div>
                                             <div data-name="dynamic_temperature" draggable="true"><span>Dynamic Temperature</span><small></small></div>
                                             <div data-name="quadratic_sampling" draggable="true"><span>Quadratic / Smooth Sampling</span><small></small></div>

                                             <div data-name="top_a" draggable="true"><span>Top A</span><small></small></div>
                                             <div data-name="min_p" draggable="true"><span>Min P</span><small></small></div>
                                             <div data-name="mirostat" draggable="true"><span>Mirostat</span><small></small></div>
 +                                            <div data-name="xtc" draggable="true"><span>XTC</span><small></small></div>
 +                                            <div data-name="encoder_repetition_penalty" draggable="true"><span>Encoder Repetition Penalty</span><small></small></div>
 +                                            <div data-name="no_repeat_ngram" draggable="true"><span>No Repeat Ngram</span><small></small></div>
                                         </div>
                                         <div id="textgenerationwebui_default_order" class="menu_button menu_button_icon">
                                             <span data-i18n="Load default order">Load default order</span>

public/scripts/textgen-settings.js+27 -0 Viewed

     'temperature',
 ];
 const OOBA_DEFAULT_ORDER = [
 +    'repetition_penalty',
 +    'presence_penalty',
 +    'frequency_penalty',
 +    'dry',
     'temperature',
     'dynamic_temperature',
     'quadratic_sampling',

     'top_a',
     'min_p',
     'mirostat',
 +    'xtc',
 +    'encoder_repetition_penalty',
 +    'no_repeat_ngram',
 ];
 const BIAS_KEY = '#textgenerationwebui_api-settings';

     });
+}
 +/**
 + * Inserts missing items from the source array into the target array.
 + * @param {any[]} source - Source array
 + * @param {any[]} target - Target array
 + * @returns {void}
 + */
 +function insertMissingArrayItems(source, target) {
 +    if (source === target || !Array.isArray(source) || !Array.isArray(target)) {
 +        return;
 +    }
++
 +    for (const item of source) {
 +        if (!target.includes(item)) {
 +            const index = source.indexOf(item);
 +            target.splice(index, 0, item);
 +        }
 +    }
 +}
++
 function setSettingByName(setting, value, trigger) {
     if (value === null || value === undefined) {
         return;

     if ('sampler_priority' === setting) {
         value = Array.isArray(value) ? value : OOBA_DEFAULT_ORDER;
 +        insertMissingArrayItems(OOBA_DEFAULT_ORDER, value);
         sortOobaItemsByOrder(value);
         settings.sampler_priority = value;
         return;