Views
No views yet
[!Warning]🚨 Qwen2.5-Math mainly supports solving English and Chinese math problems through CoT and TIR. We do not recommend using this series of models for other tasks.

transformers>=4.37.0 for Qwen2.5-Math models. The latest version is recommended.[!Warning]🚨 This is a must becausetransformersintegrated Qwen2 codes since4.37.0.
[!Important]Qwen2.5-Math-1.5B-Instruct is an instruction model for chatting;Qwen2.5-Math-1.5B is a base model typically used for completion and few-shot inference, serving as a better starting point for fine-tuning.
transformers:1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "Qwen/Qwen2.5-Math-1.5B-Instruct"
4device = "cuda" # the device to load the model onto
5
6model = AutoModelForCausalLM.from_pretrained(
7 model_name,
8 torch_dtype="auto",
9 device_map="auto"
10)
11tokenizer = AutoTokenizer.from_pretrained(model_name)
12
13prompt = "Find the value of $x$ that satisfies the equation $4x+5 = 6x+7$."
14
15# CoT
16messages = [
17 {"role": "system", "content": "Please reason step by step, and put your final answer within \\boxed{}."},
18 {"role": "user", "content": prompt}
19]
20
21# TIR
22messages = [
23 {"role": "system", "content": "Please integrate natural language reasoning with programs to solve the problem above, and put your final answer within \\boxed{}."},
24 {"role": "user", "content": prompt}
25]
26
27text = tokenizer.apply_chat_template(
28 messages,
29 tokenize=False,
30 add_generation_prompt=True
31)
32model_inputs = tokenizer([text], return_tensors="pt").to(device)
33
34generated_ids = model.generate(
35 **model_inputs,
36 max_new_tokens=512
37)
38generated_ids = [
39 output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
40]
41
42response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]@article{yang2024qwen25mathtechnicalreportmathematical,
title={Qwen2.5-Math Technical Report: Toward Mathematical Expert Model via Self-Improvement},
author={An Yang and Beichen Zhang and Binyuan Hui and Bofei Gao and Bowen Yu and Chengpeng Li and Dayiheng Liu and Jianhong Tu and Jingren Zhou and Junyang Lin and Keming Lu and Mingfeng Xue and Runji Lin and Tianyu Liu and Xingzhang Ren and Zhenru Zhang},
journal={arXiv preprint arXiv:2409.12122},
year={2024}
}