From 2a6236a2eccafe1f5a16b77c486ec132b9a8e96b Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Tue, 30 Aug 2022 15:45:18 -0400
Subject: [PATCH] Fix for united new sampler order

---
 static/koboldai.js             | 2 ++
 templates/settings flyout.html | 1 +
 2 files changed, 3 insertions(+)

diff --git a/static/koboldai.js b/static/koboldai.js
index 42799f46..b7e113ff 100644
--- a/static/koboldai.js
+++ b/static/koboldai.js
@@ -44,6 +44,7 @@ map1.set('Top P Sampling', 2)
 map1.set('Tail Free Sampling', 3)
 map1.set('Typical Sampling', 4)
 map1.set('Temperature', 5)
+map1.set('Repetition Penalty', 6)
 const map2 = new Map()
 map2.set(0, 'Top K Sampling')
 map2.set(1, 'Top A Sampling')
@@ -51,6 +52,7 @@ map2.set(2, 'Top P Sampling')
 map2.set(3, 'Tail Free Sampling')
 map2.set(4, 'Typical Sampling')
 map2.set(5, 'Temperature')
+map2.set(6, 'Repetition Penalty')
 var calc_token_usage_timeout;
 var game_text_scroll_timeout;
 var var_processing_time = 0;
diff --git a/templates/settings flyout.html b/templates/settings flyout.html
index c4fb2899..e2452d17 100644
--- a/templates/settings flyout.html	
+++ b/templates/settings flyout.html	
@@ -114,6 +114,7 @@
 					<span class="setting_item">
 						<div style="display:flex;flex-direction:row;">
 							<ul id="sample_order_list" style="width:calc(var(--flyout_menu_width) - 60px);list-style-position: inside; padding: 0;">
+								<li class="sample_order cursor" onclick="select_sample(this);">Repetition Penalty</li>
 								<li class="sample_order cursor" onclick="select_sample(this);">Top K Sampling</li>
 								<li class="sample_order cursor" onclick="select_sample(this);">Top A Sampling</li>
 								<li class="sample_order cursor" onclick="select_sample(this);">Top P Sampling</li>