summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorPierrick Hymbert <pierrick.hymbert@gmail.com>2024-04-26 09:26:16 +0200
committerGitHub <noreply@github.com>2024-04-26 09:26:16 +0200
commit5790c8dac1a4f0aa80b4efee3b962d8c04c829e8 (patch)
tree1edb716d2be1c5fb92dc3a425882e3aa46dffff7
parent46e12c4692a37bdd31a0432fc5153d7d22bc7f72 (diff)
bench: server add stop word for PHI-2 (#6916)
-rw-r--r--examples/server/bench/script.js3
1 files changed, 2 insertions, 1 deletions
diff --git a/examples/server/bench/script.js b/examples/server/bench/script.js
index c4c486cd..bdf4f5ab 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)};