diff options
author | Pierrick Hymbert <pierrick.hymbert@gmail.com> | 2024-04-26 09:26:16 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-04-26 09:26:16 +0200 |
commit | 5790c8dac1a4f0aa80b4efee3b962d8c04c829e8 (patch) | |
tree | 1edb716d2be1c5fb92dc3a425882e3aa46dffff7 /examples/server | |
parent | 46e12c4692a37bdd31a0432fc5153d7d22bc7f72 (diff) |
bench: server add stop word for PHI-2 (#6916)
Diffstat (limited to 'examples/server')
-rw-r--r-- | examples/server/bench/script.js | 3 |
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)}; |