[Fix] update interlm2 chat template(#1002)

InternLM · Jan 19, 2024 · c8dad23 · c8dad23
1 parent d2efef3
commit c8dad23
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/lmdeploy/model.py b/lmdeploy/model.py
@@ -338,13 +338,13 @@ class InternLM2Chat7B(InternLMChat7B):
 
     def __init__(self,
                  session_len=32768,
-                 system='[UNUSED_TOKEN_146]system\n',
-                 user='[UNUSED_TOKEN_146]user\n',
-                 assistant='[UNUSED_TOKEN_146]assistant\n',
-                 eosys='[UNUSED_TOKEN_145]\n',
-                 eoh='[UNUSED_TOKEN_145]\n',
-                 eoa='[UNUSED_TOKEN_145]\n',
-                 stop_words=['[UNUSED_TOKEN_145]'],
+                 system='<|im_start|>system\n',
+                 user='<|im_start|>user\n',
+                 assistant='<|im_start|>assistant\n',
+                 eosys='<|im_end|>\n',
+                 eoh='<|im_end|>\n',
+                 eoa='<|im_end|>\n',
+                 stop_words=['<|im_end|>', '<|action_end|>'],
                  **kwargs):
         super(InternLM2Chat7B, self).__init__(session_len=session_len,
                                               system=system,