update templates

InternLM · Oct 9, 2023 · eae7da8 · eae7da8
1 parent 3a08bd6
commit eae7da8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/xtuner/utils/templates.py b/xtuner/utils/templates.py
@@ -2,11 +2,14 @@
 from mmengine.config import ConfigDict
 
 PROMPT_TEMPLATE = ConfigDict(
+    default=dict(
+        SYSTEM='<|System|>:{system}\n',
+        INSTRUCTION='<|User|>:{input}\n<|Bot|>:'),
     internlm_chat=dict(
         SYSTEM='<|System|>:{system}\n',
         INSTRUCTION='<|User|>:{input}<eoh>\n<|Bot|>:'),
     moss_sft=dict(SYSTEM='{system}\n', INSTRUCTION='<|Human|>: {input}<eoh>'),
-    llama_2_chat=dict(
+    llama2_chat=dict(
         SYSTEM=(
             '[INST] <<SYS>>\n You are a helpful, respectful and honest '
             'assistant. Always answer as helpfully as possible, while being '