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-7B-Instruct is an instruction model for chatting;Qwen2.5-Math-7B is a base model typically used for completion and few-shot inference, serving as a better starting point for fine-tuning.
transformers:1from modelscope import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "qwen/Qwen2.5-Math-7B-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$."
14messages = [
15 {"role": "system", "content": "You are Qwen, created by Alibaba Cloud. You are a helpful assistant."},
16 {"role": "user", "content": prompt}
17]
18text = tokenizer.apply_chat_template(
19 messages,
20 tokenize=False,
21 add_generation_prompt=True
22)
23model_inputs = tokenizer([text], return_tensors="pt").to(device)
24
25generated_ids = model.generate(
26 **model_inputs,
27 max_new_tokens=512
28)
29generated_ids = [
30 output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
31]
32
33response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]snapshot_download can help you solve issues concerning downloading checkpoints.@article{yang2024qwen2,
title={Qwen2 technical report},
author={Yang, An and Yang, Baosong and Hui, Binyuan and Zheng, Bo and Yu, Bowen and Zhou, Chang and Li, Chengpeng and Li, Chengyuan and Liu, Dayiheng and Huang, Fei and others},
journal={arXiv preprint arXiv:2407.10671},
year={2024}
}