handle if assistant_message_gen and assistant_message_gen!=assistant_message_start, replace final output tag with unspaced (gen) version if exists

This commit is contained in:
Concedo
2025-08-10 16:51:34 +08:00
parent 204739e7f1
commit 8e6d27f629
4 changed files with 26 additions and 2 deletions
+9
View File
@@ -0,0 +1,9 @@
{
"system_start": "<|bom|><|system|> ",
"system_end": "<|eom|>",
"user_start": "<|bom|><|user|> ",
"user_end": "<|eom|>",
"assistant_start": "<|bom|><|assistant|> ",
"assistant_gen": "<|bom|><|assistant|>",
"assistant_end": "<|eom|>"
}
+2 -1
View File
@@ -3,6 +3,7 @@
"system_end": "",
"user_start": "[INST] ",
"user_end": "",
"assistant_start": "[/INST]",
"assistant_start": "[/INST] ",
"assistant_gen": "[/INST]",
"assistant_end": "</s>"
}
+2 -1
View File
@@ -3,6 +3,7 @@
"system_end": "\n\n",
"user_start": "User: ",
"user_end": "\n\n",
"assistant_start": "Assistant:",
"assistant_start": "Assistant: ",
"assistant_gen": "Assistant:",
"assistant_end": "\n\n"
}