support Yi-34B-Chat models

Former-commit-id: ff1c289229

src/llmtuner/data/template.py

@@ -682,6 +682,22 @@ register_template(
 )
 
 
+register_template(
+    name="yi",
+    prefix=[
+        "{{system}}"
+    ],
+    prompt=[
+        "<|im_start|>user\n{{query}}<|im_end|>\n<|im_start|>assistant\n"
+    ],
+    system="",
+    sep=[
+        "<|im_end|>\n"
+    ],
+    efficient_eos=True
+)
+
+
 register_template(
     name="zephyr",
     prefix=[