Add EOS token to the generation config
Browse files(Same as PR#100, but for the generation config)
- generation_config.json +3 -1
generation_config.json
CHANGED
|
@@ -1,4 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"_from_model_config": true,
|
| 3 |
-
"transformers_version": "4.37.0.dev0"
|
|
|
|
|
|
|
| 4 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"_from_model_config": true,
|
| 3 |
+
"transformers_version": "4.37.0.dev0",
|
| 4 |
+
"eos_token_id": 50256,
|
| 5 |
+
"bos_token_id": 50256
|
| 6 |
}
|