Fix special tokens for generation (#4)
Browse files- Fix special tokens for generation (dbdd99c51d4099c8d3a8fbe89805cde363327383)
Co-authored-by: Younes B <[email protected]>
- generation_config.json +1 -1
generation_config.json
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
{
|
| 2 |
"_from_model_config": true,
|
| 3 |
"bos_token_id": 1,
|
| 4 |
-
"eos_token_id": 11,
|
| 5 |
"pad_token_id": 0,
|
| 6 |
"transformers_version": "4.52.0.dev0"
|
| 7 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"_from_model_config": true,
|
| 3 |
"bos_token_id": 1,
|
| 4 |
+
"eos_token_id": [11, 228],
|
| 5 |
"pad_token_id": 0,
|
| 6 |
"transformers_version": "4.52.0.dev0"
|
| 7 |
}
|