Skip to content

Commit 812f4b7

Browse files
author
Tong Li
committed
update loader
1 parent 7f2ceac commit 812f4b7

File tree

1 file changed

+1
-1
lines changed
  • applications/ColossalChat/coati/dataset

1 file changed

+1
-1
lines changed

applications/ColossalChat/coati/dataset/loader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -357,7 +357,7 @@ def apply_chat_template_and_mask(
357357
ignore_idx: int = -100,
358358
) -> Dict[str, torch.Tensor]:
359359

360-
system_prompt = "You are a helpful assistant. The assistant first thinks about the reasoning process in the mind and then provides the user with the answer. The reasoning process and answer are enclosed within <think> </think> and<answer> </answer> tags, respectively, i.e., <think> reasoning process here </think><answer> answer here </answer>. Now the user asks you to solve a math problem that involves reasoning. After thinking, when you finally reach a conclusion, clearly output the final answer without explanation within the <answer> </answer> tags, your final answer should be a integer without unit, currency mark, thousands separator or other text. i.e., <answer> 123 </answer>.\n"
360+
system_prompt = "You are a helpful assistant. The assistant first thinks about the reasoning process in the mind and then provides the user with the answer. The reasoning process and answer are enclosed within <think> </think> and<answer> </answer> tags, respectively, i.e., <think> reasoning process here </think><answer> answer here </answer>. Now the user asks you to solve a math problem that involves reasoning.\n"
361361

362362
system_element = {
363363
"role": "system",

0 commit comments

Comments
 (0)