From 5790c8dac1a4f0aa80b4efee3b962d8c04c829e8 Mon Sep 17 00:00:00 2001
From: Pierrick Hymbert <pierrick.hymbert@gmail.com>
Date: Fri, 26 Apr 2024 09:26:16 +0200
Subject: [PATCH] bench: server add stop word for PHI-2 (#6916)

---
 examples/server/bench/script.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples/server/bench/script.js b/examples/server/bench/script.js
index c4c486cdf..bdf4f5abc 100644
--- a/examples/server/bench/script.js
+++ b/examples/server/bench/script.js
@@ -90,7 +90,8 @@ export default function () {
         "model": model,
         "stream": true,
         "seed": 42,
-        "max_tokens": max_tokens
+        "max_tokens": max_tokens,
+        "stop": ["<|im_end|>"] // This is temporary for phi-2 base (i.e. not instructed) since the server expects that the model always to emit BOS
     }
 
     const params = {method: 'POST', body: JSON.stringify(payload)};