add models

Former-commit-id: 0fdf61b2f765c125acda4f406eb25b3e59e75db2
2024-02-06 14:57:23 +08:00
parent 73f40f1ca4
commit 0dd68d1e06
2 changed files with 16 additions and 20 deletions
--- a/src/llmtuner/data/template.py
+++ b/src/llmtuner/data/template.py
@@ -376,7 +376,8 @@ register_template(

 register_template(
    name="deepseekcoder",
-    format_user=StringFormatter(slots=["### Instruction:\n{{content}}\n### Response:\n"]),
+    format_user=StringFormatter(slots=["### Instruction:\n{{content}}\n### Response:"]),
+    format_assistant=StringFormatter(slots=["\n", "{{content}}"]),
    format_separator=EmptyFormatter(slots=["\n", {"token": "<|EOT|>"}, "\n"]),
    default_system=(
        "You are an AI programming assistant, utilizing the Deepseek Coder model, "