diff --git a/lmdeploy/model.py b/lmdeploy/model.py index 19043e485e..22f55f5203 100644 --- a/lmdeploy/model.py +++ b/lmdeploy/model.py @@ -334,13 +334,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,