From 4efb75d131c48aaee158de77bc8698c7230e21b3 Mon Sep 17 00:00:00 2001 From: ZHU QIHAO <18811325956@163.com> Date: Tue, 6 Feb 2024 19:24:28 +0800 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 38ee5f3..4e364a3 100644 --- a/README.md +++ b/README.md @@ -170,7 +170,7 @@ model.generation_config = GenerationConfig.from_pretrained(model_name) model.generation_config.pad_token_id = model.generation_config.eos_token_id messages = [ - {"role": "user", "content": "what is the integral of x^2 from 0 to 2?"} + {"role": "user", "content": "what is the integral of x^2 from 0 to 2?\nPlease reason step by step, and put your final answer within \boxed{}."} ] input_tensor = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt") outputs = model.generate(input_tensor.to(model.device), max_new_tokens=100)