diff --git a/public/index.html b/public/index.html
index 02bcc1f0f..bd5c12692 100644
--- a/public/index.html
+++ b/public/index.html
@@ -547,6 +547,36 @@
+
No Repeat Ngram Size
diff --git a/public/scripts/textgen-settings.js b/public/scripts/textgen-settings.js
index 4dcfe091c..9d9e9e5bc 100644
--- a/public/scripts/textgen-settings.js
+++ b/public/scripts/textgen-settings.js
@@ -44,6 +44,8 @@ const textgenerationwebui_settings = {
length_penalty: 1,
min_length: 0,
encoder_rep_pen: 1,
+ freq_pen: 0,
+ presence_pen: 0,
do_sample: true,
early_stopping: false,
seed: -1,
@@ -87,6 +89,8 @@ const setting_names = [
"length_penalty",
"min_length",
"encoder_rep_pen",
+ "freq_pen",
+ "presence_pen",
"do_sample",
"early_stopping",
"seed",
@@ -413,6 +417,8 @@ export function getTextGenGenerationData(finalPrompt, this_amount_gen, isImperso
'repetition_penalty': textgenerationwebui_settings.rep_pen,
'repetition_penalty_range': textgenerationwebui_settings.rep_pen_range,
'encoder_repetition_penalty': textgenerationwebui_settings.encoder_rep_pen,
+ 'frequency_penalty': textgenerationwebui_settings.freq_pen,
+ 'presence_penalty': textgenerationwebui_settings.presence_pen,
'top_k': textgenerationwebui_settings.top_k,
'min_length': textgenerationwebui_settings.min_length,
'no_repeat_ngram_size': textgenerationwebui_settings.no_repeat_ngram_size,