x54-729 commited on
Commit
41408e7
1 Parent(s): e338c92

Add special tokens for chat

Browse files
Files changed (2) hide show
  1. special_tokens_map.json +8 -0
  2. tokenizer_config.json +8 -0
special_tokens_map.json CHANGED
@@ -1,4 +1,12 @@
1
  {
 
 
 
 
 
 
 
 
2
  "bos_token": "<s>",
3
  "eos_token": "</s>",
4
  "pad_token": "</s>",
 
1
  {
2
+ "additional_special_tokens": [
3
+ "<|im_start|>",
4
+ "<|im_end|>",
5
+ "<|action_start|>",
6
+ "<|action_end|>",
7
+ "<|interpreter|>",
8
+ "<|plugin|>"
9
+ ],
10
  "bos_token": "<s>",
11
  "eos_token": "</s>",
12
  "pad_token": "</s>",
tokenizer_config.json CHANGED
@@ -86,5 +86,13 @@
86
  "special": true
87
  }
88
  },
 
 
 
 
 
 
 
 
89
  "chat_template": "{{ bos_token }}{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}"
90
  }
 
86
  "special": true
87
  }
88
  },
89
+ "additional_special_tokens": [
90
+ "<|im_start|>",
91
+ "<|im_end|>",
92
+ "<|action_start|>",
93
+ "<|action_end|>",
94
+ "<|interpreter|>",
95
+ "<|plugin|>"
96
+ ],
97
  "chat_template": "{{ bos_token }}{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}"
98
  }