bobber commited on
Commit
50ce4f4
·
verified ·
1 Parent(s): 03fe998

fix: add token 11 (im_end) to eos_token_id — stops generation at end of turn

Browse files
Files changed (1) hide show
  1. generation_config.json +5 -2
generation_config.json CHANGED
@@ -1,8 +1,11 @@
1
  {
2
  "_from_model_config": true,
3
  "bos_token_id": 1,
4
- "eos_token_id": 2,
 
 
 
5
  "max_length": 262144,
6
  "pad_token_id": 0,
7
  "transformers_version": "5.3.0"
8
- }
 
1
  {
2
  "_from_model_config": true,
3
  "bos_token_id": 1,
4
+ "eos_token_id": [
5
+ 2,
6
+ 11
7
+ ],
8
  "max_length": 262144,
9
  "pad_token_id": 0,
10
  "transformers_version": "5.3.0"
11
+ }