From cd8dd5fc8a7e8a6421570128b6f0a7fe98a54a57 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Sat, 31 Aug 2024 20:31:34 +0300
Subject: [PATCH] XTC for ooba

---
 public/index.html                  | 3 ++-
 public/scripts/textgen-settings.js | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/public/index.html b/public/index.html
index 4403372bf..eb0bcee4a 100644
--- a/public/index.html
+++ b/public/index.html
@@ -1279,7 +1279,7 @@
                                         </div>
                                     </div>
 
-                                    <div data-newbie-hidden data-tg-type="koboldcpp" id="xtc_block" class="wide100p">
+                                    <div data-newbie-hidden data-tg-type="koboldcpp,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">
@@ -1649,6 +1649,7 @@
                                             <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>
                                         <div id="textgenerationwebui_default_order" class="menu_button menu_button_icon">
                                             <span data-i18n="Load default order">Load default order</span>
diff --git a/public/scripts/textgen-settings.js b/public/scripts/textgen-settings.js
index 161182381..b6786e768 100644
--- a/public/scripts/textgen-settings.js
+++ b/public/scripts/textgen-settings.js
@@ -80,6 +80,7 @@ const OOBA_DEFAULT_ORDER = [
     'top_a',
     'min_p',
     'mirostat',
+    'xtc',
 ];
 const BIAS_KEY = '#textgenerationwebui_api-settings';