Add EOS token to the generation config (#101)
Browse files- Add EOS token to the generation config (9a5117d15e407c48fa40f8adbac1244abae52d6f)
Co-authored-by: Joao Gante <[email protected]>
- 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 |
}
|