Update README.md

README.md

@@ -66,7 +66,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 instruction = "Please reason step by step, and put your final answer within \\boxed{}."
 prompt = "The arithmetic mean of 7, 2, $x$ and 10 is 9. What is the value of $x$?"
-#prompt = "What is the smallest positive perfect cube that can be written as the sum of three consecutive integers?"
+# prompt = "What is the smallest positive perfect cube that can be written as the sum of three consecutive integers?"
 # prompt = "How many r's are in the word \"strawberry\""
 
 messages = [
@@ -79,6 +79,8 @@ text = tokenizer.apply_chat_template(
     add_generation_prompt=True
 )
 
+# text = text + "<think>" # Uncomment this to force thinking mode
+
 model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
 
 generated_ids = model.generate(