Update README.md

README.md

@@ -71,17 +71,21 @@ You can format it into this InternLM2 Dialogue format like,
 ```
 def InternLM2_format(instruction,prompt,answer,history):
     prefix_template=[
-        "<|system|>:",
-        "{}"
+        "<|im_start|>system\n",
+        "{}",
+        "<|im_end|>\n"
     ]
     prompt_template=[
-        "<|user|>:",
-        "{}\n",
-        "<|Bot|>:\n"
+        "<|im_start|>user\n",
+        "{}",
+        "<|im_end|>\n"
+        "<|im_start|>assistant\n",
+        "{}",
+        "<|im_end|>\n"
     ]
-    system = f'{prefix_template[0]}\n{prefix_template[-1].format(instruction)}\n'
-    history = "\n".join([f'{prompt_template[0]}\n{prompt_template[1].format(qa[0])}{prompt_template[-1]}{qa[1]}' for qa in history])
-    prompt = f'\n{prompt_template[0]}\n{prompt_template[1].format(prompt)}{prompt_template[-1]}'
+    system = f'{prefix_template[0]}{prefix_template[1].format(instruction)}{prefix_template[2]}'
+    history = "".join([f'{prompt_template[0]}{prompt_template[1].format(qa[0])}{prompt_template[2]}{prompt_template[3]}{prompt_template[4].format(qa[1])}{prompt_template[5]}' for qa in history])
+    prompt = f'{prompt_template[0]}{prompt_template[1].format(prompt)}{prompt_template[2]}{prompt_template[3]}'
     return f"{system}{history}{prompt}"
 ```
 And there is a good example for system prompt,