Update README.md

README.md

@@ -66,15 +66,15 @@ For queries in ShareGPT format like,
 You can format it into this InternLM2 Dialogue format like,
 ```
 def InternLM2_format(instruction,prompt,answer,history):
-    prefix_template=[
-            "[UNUSED_TOKEN_146]system\n",
-            "{}[UNUSED_TOKEN_145]\n"
-        ]
-    prompt_template=[
-            "[UNUSED_TOKEN_146]user\n",
-            "{}[UNUSED_TOKEN_145]\n",
-            "[UNUSED_TOKEN_146]assistant\n"
-        ]
+    prefix=[
+        "<|system|>:",
+        "{}"
+    ]
+    prompt=[
+        "<|user|>:",
+        "{}\n",
+        "<|Bot|>:\n"
+    ]
     system = f'{prefix_template[0]}{prefix_template[-1].format(instruction)}'
     history = "\n".join([f'{prompt_template[0]}{prompt_template[1].format(qa[0])}\n{prompt_template[-1]}{qa[1]}' for qa in history])
     prompt = f'\n{prompt_template[0]}{prompt_template[1].format(prompt)}\n{prompt_template[-1]}'