change chat template (#216)

C1rN09 · web-flow · commit 192646795ed9 · 2024-01-23T12:36:16.000-08:00
diff --git a/src/alpaca_eval/models_configs/internlm2-chat-20b-ppo/configs.yaml b/src/alpaca_eval/models_configs/internlm2-chat-20b-ppo/configs.yaml
@@ -10,6 +10,6 @@ internlm2-chat-20b-ppo:
     max_new_tokens: 2048
     batch_size: 32
     eos_token_id: 92542
-    remove_ending: "[UNUSED_TOKEN_145]"
+    remove_ending: "<|im_end|>"
   pretty_name: "InternLM2 Chat 20B"
   link: "https://huggingface.co/internlm/internlm2-chat-20b"
diff --git a/src/alpaca_eval/models_configs/internlm2-chat-20b-ppo/prompt.txt b/src/alpaca_eval/models_configs/internlm2-chat-20b-ppo/prompt.txt
@@ -1,3 +1,3 @@
-[UNUSED_TOKEN_146]user
-{instruction}[UNUSED_TOKEN_145]
-[UNUSED_TOKEN_146]assistant
+<|im_start|>user
+{instruction}<|im_end|>
+<|im_start|>assistant
