Skip to content

Commit a81e4a8

Browse files
committed
Remake chat template test
1 parent f6cc0c1 commit a81e4a8

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/test-chat-template.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -292,11 +292,11 @@ int main(void) {
292292
},
293293
{
294294
/* .name= */ "ByteDance-Seed/Seed-OSS-36B-Instruct",
295-
/* .template_str */ "{%- for message in messages %}{%- if message.role in [\"user\", \"system\"] %}{{ '<seed:bos>' + message.role + \"\\n\" + message.content + '<seed:eos>' }}{%- elif message.role == \"assistant\" %}{{ '<seed:bos>' + message.role }}{%- if message.content is defined and message.content is string and message.content|trim|length > 0 %}{{ \"\n\" + message.content|trim + '<seed:eos>' }}{%- endif %}{%- else %}{{ '<seed:bos>' + message.role + \"\n\" + message.content + '<seed:eos>' }}{%- endif %}{%- endfor %}{%- if add_generation_prompt %}{{ '<seed:bos>' + \"assistant\\n\" }}{%- endif %}",
295+
/* .template_str */ "{# <seed:bos> #}{%- for message in messages %}{%- if message.role in [\"user\", \"system\"] %}{{ bos_token + message.role + \"\\n\" + message.content + eos_token }}{%- elif message.role == \"assistant\" %}{{ bos_token + message.role }}{%- if message.content is defined and message.content is string and message.content|trim|length > 0 %}{{ \"\n\" + message.content|trim + eos_token }}{%- endif %}{%- else %}{{ bos_token + message.role + \"\n\" + message.content + eos_token }}{%- endif %}{%- endfor %}{%- if add_generation_prompt %}{{ bos_token + \"assistant\\n\" }}{%- endif %}",
296296
/* .expected_output= */ "<seed:bos>system\nYou are a helpful assistant<seed:eos><seed:bos>user\nHello<seed:eos><seed:bos>assistant\nHi there<seed:eos><seed:bos>user\nWho are you<seed:eos><seed:bos>assistant\n I am an assistant <seed:eos><seed:bos>user\nAnother question<seed:eos><seed:bos>assistant",
297297
/* .expected_output_jinja= */ "<seed:bos>system\nYou are a helpful assistant<seed:eos><seed:bos>user\nHello<seed:eos><seed:bos>assistant\nHi there<seed:eos><seed:bos>user\nWho are you<seed:eos><seed:bos>assistant\nI am an assistant<seed:eos><seed:bos>user\nAnother question<seed:eos><seed:bos>assistant\n",
298-
/* .bos_token= */ "",
299-
/* .eos_token= */ "",
298+
/* .bos_token= */ "<seed:bos>",
299+
/* .eos_token= */ "<seed:eos>",
300300
}
301301
};
302302
std::vector<char> formatted_chat(1024);

0 commit comments

Comments
 (0)