From 50de67898020810e6d79fc62c1102be91f3d77ac Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Sat, 14 Sep 2024 16:53:21 +0300
Subject: [PATCH] Hide beam search for vllm. It never worked.

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

diff --git a/public/index.html b/public/index.html
index e08105239..dd4a111e8 100644
--- a/public/index.html
+++ b/public/index.html
@@ -1396,7 +1396,7 @@
                                             </div>
                                         </div>
                                     </div>
-                                    <div data-tg-type="ooba, vllm" id="beamSearchBlock" name="beamSearchBlock" class="wide100p">
+                                    <div data-tg-type="ooba" id="beamSearchBlock" name="beamSearchBlock" class="wide100p">
                                         <h4 class="wide100p textAlignCenter">
                                             <label>
                                                 <span data-i18n="Beam search">Beam Search</span>
diff --git a/public/scripts/textgen-settings.js b/public/scripts/textgen-settings.js
index 1fb5917e7..c22ecee23 100644
--- a/public/scripts/textgen-settings.js
+++ b/public/scripts/textgen-settings.js
@@ -1120,8 +1120,8 @@ export function getTextGenGenerationData(finalPrompt, maxTokens, isImpersonate,
         'minimum_message_content_tokens': settings.type === DREAMGEN ? settings.min_length : undefined,
         'min_tokens': settings.min_length,
         'num_beams': settings.type === OOBA ? settings.num_beams : undefined,
-        'length_penalty': settings.length_penalty,
-        'early_stopping': settings.early_stopping,
+        'length_penalty': settings.type === OOBA ? settings.length_penalty : undefined,
+        'early_stopping': settings.type === OOBA ? settings.early_stopping : undefined,
         'add_bos_token': settings.add_bos_token,
         'dynamic_temperature': dynatemp ? true : undefined,
         'dynatemp_low': dynatemp ? settings.min_temp : undefined,