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

2025-12-16 11:50:35 +08:00 · 2023-08-09 16:23:31 +08:00
parent 2d90685358
commit 39cd8b6989
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"