Update generation_config.json
Browse files- generation_config.json +4 -4
generation_config.json
CHANGED
@@ -1,11 +1,11 @@
|
|
1 |
{
|
2 |
"chat_format": "chatml",
|
3 |
-
"do_sample":
|
4 |
"eos_token_id": 151643,
|
5 |
-
"max_new_tokens":
|
6 |
"max_window_size": 6144,
|
7 |
"pad_token_id": 151643,
|
8 |
-
"top_k":
|
9 |
-
"top_p": 0.
|
10 |
"transformers_version": "4.31.0"
|
11 |
}
|
|
|
1 |
{
|
2 |
"chat_format": "chatml",
|
3 |
+
"do_sample": false, // Disable sampling for faster inference
|
4 |
"eos_token_id": 151643,
|
5 |
+
"max_new_tokens": 256, // Reduce the number of tokens to generate
|
6 |
"max_window_size": 6144,
|
7 |
"pad_token_id": 151643,
|
8 |
+
"top_k": 50, // Use a reasonable value for top_k
|
9 |
+
"top_p": 0.9, // Increase top_p to allow more tokens
|
10 |
"transformers_version": "4.31.0"
|
11 |
}
|