Update README.md

README.md

@@ -59,14 +59,14 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 question = "Find the value of $x$ that satisfies the equation $4x+5 = 6x+7$."
 
-sys_prompt="<|im_start|>system\nPlease reason step by step, and put your final answer within \\boxed{{}}.<|im_end|>\n"
-query_prompt="<|im_start|>user" + "\n"
+sys_prompt = "<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n"
+query_prompt = "<|im_start|>user" + "\n"
 # {query}
-prompt_after_query="<|im_end|>" + "\n"
-resp_prompt="<|im_start|>assistant" + "\n"
-prompt_before_resp=""
+prompt_after_query = "\n" + "Please reason step by step, and put your final answer within \\boxed{}.<|im_end|>" + "\n"
+resp_prompt = "<|im_start|>assistant" + "\n"
+prompt_before_resp = ""
 # {resp}
-delim="<|im_end|>" + "\n"
+delim = "<|im_end|>" + "\n"
 
 prefix_prompt = f"{query_prompt}{question}{prompt_after_query}{resp_prompt}{prompt_before_resp}".rstrip(" ")
 full_prompt = sys_prompt + delim.join([prefix_prompt])