From a5b5dfdcb62958ca5d3e3ab5a64a1467a9607187 Mon Sep 17 00:00:00 2001
From: RossAscends <124905043+RossAscends@users.noreply.github.com>
Date: Sun, 5 Nov 2023 23:50:07 +0900
Subject: [PATCH] temperature last for ooba

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

diff --git a/public/index.html b/public/index.html
index abf9ba3d0..d36c7a094 100644
--- a/public/index.html
+++ b/public/index.html
@@ -1383,6 +1383,12 @@
                                                 <input type="checkbox" id="skip_special_tokens_textgenerationwebui" />
                                                 <small data-i18n="Skip Special Tokens">Skip Special Tokens</small>
                                             </label>
+                                            <label data-forAphro=False class="checkbox_label" for="temperature_last_textgenerationwebui">
+                                                <input type="checkbox" id="temperature_last_textgenerationwebui" />
+                                                <small data-i18n="Temperature Last">Temperature Last
+                                                    <div class="fa-solid fa-circle-info opacity50p " data-i18n="Use the temperature sampler last." title="Use the temperature sampler last."></div>
+                                                </small>
+                                            </label>
                                             <!--
                                             <label data-forAphro=True class="checkbox_label" for="spaces_between_special_tokens_aphrodite_textgenerationwebui">
                                                 <input type="checkbox" id="spaces_between_special_tokens_aphrodite_textgenerationwebui" />
diff --git a/public/scripts/textgen-settings.js b/public/scripts/textgen-settings.js
index a5f2b7f07..55dedefc9 100644
--- a/public/scripts/textgen-settings.js
+++ b/public/scripts/textgen-settings.js
@@ -29,6 +29,7 @@ export const textgen_types = {
 
 const textgenerationwebui_settings = {
     temp: 0.7,
+    temperature_last: true,
     top_p: 0.5,
     top_k: 40,
     top_a: 0,
@@ -82,6 +83,7 @@ export let textgenerationwebui_preset_names = [];
 
 const setting_names = [
     "temp",
+    "temperature_last",
     "rep_pen",
     "rep_pen_range",
     "no_repeat_ngram_size",
@@ -456,6 +458,7 @@ export function getTextGenGenerationData(finalPrompt, this_amount_gen, isImperso
         'max_new_tokens': this_amount_gen,
         'do_sample': textgenerationwebui_settings.do_sample,
         'temperature': textgenerationwebui_settings.temp,
+        'temperature_last': textgenerationwebui_settings.temperature_last,
         'top_p': textgenerationwebui_settings.top_p,
         'typical_p': textgenerationwebui_settings.typical_p,
         'min_p': textgenerationwebui_settings.min_p,