fix: stop generation at eos exactly like the original model

#4
Files changed (1) hide show
  1. generation_config.json +5 -1
generation_config.json CHANGED
@@ -1,7 +1,11 @@
1
  {
2
  "_from_model_config": true,
3
  "bos_token_id": 1,
4
- "eos_token_id": 32000,
 
 
 
 
5
  "pad_token_id": 32000,
6
  "transformers_version": "4.41.0.dev0"
7
  }
 
1
  {
2
  "_from_model_config": true,
3
  "bos_token_id": 1,
4
+ "eos_token_id": [
5
+ 32000,
6
+ 32001,
7
+ 32007
8
+ ],
9
  "pad_token_id": 32000,
10
  "transformers_version": "4.41.0.dev0"
11
  }