Update README.md
Browse files
README.md
CHANGED
|
@@ -69,16 +69,7 @@ def generate(question_list,model_path):
|
|
| 69 |
return completions
|
| 70 |
|
| 71 |
def prepare_prompt(question, tokenizer):
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
msg = [
|
| 75 |
-
{"role": "user", "content": content}
|
| 76 |
-
]
|
| 77 |
-
prompt = tokenizer.apply_chat_template(
|
| 78 |
-
msg,
|
| 79 |
-
tokenize=False,
|
| 80 |
-
add_generation_prompt=True
|
| 81 |
-
)
|
| 82 |
return prompt
|
| 83 |
|
| 84 |
def run():
|
|
|
|
| 69 |
return completions
|
| 70 |
|
| 71 |
def prepare_prompt(question, tokenizer):
|
| 72 |
+
prompt = f"<|im_start|>user\nSolve the following math problem efficiently and clearly.\nPlease reason step by step, and put your final answer within \\boxed{{}}.\nProblem: {question}<|im_end|>\n<|im_start|>assistant\n"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 73 |
return prompt
|
| 74 |
|
| 75 |
def run():
|