fix rm #420, fix template #426, fix #423

Former-commit-id: 70ea3caaa7a7695c77179cd1bb18707a80a373d7
2023-08-09 16:23:31 +08:00
parent 031a819257
commit a3a7465f00
3 changed files with 7 additions and 10 deletions
--- a/src/llmtuner/extras/template.py
+++ b/src/llmtuner/extras/template.py
@@ -388,12 +388,9 @@ register_template(
    name="intern",
    prefix=[],
    prompt=[
-        {"token": "<|User|>"},
-        ":{{query}}",
+        "<|User|>:{{query}}",
        {"token": "<eoh>"},
-        "\n",
-        {"token": "<|Bot|>"},
-        ":"
+        "\n<|Bot|>:"
    ],
    sep=[
        "\n"