keminglu commited on
Commit
75765db
·
verified ·
1 Parent(s): b294907

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +100 -0
README.md ADDED
@@ -0,0 +1,100 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - chat
8
+ ---
9
+
10
+
11
+ # Qwen2-Math-7B-Instruct
12
+
13
+ ## Introduction
14
+
15
+ Over the past year, we have dedicated significant effort to researching and enhancing the reasoning capabilities of large language models, with a particular focus on their ability to solve arithmetic and mathematical problems. Today, we are delighted to introduce a serise of math-specific large language models of our Qwen2 series, Qwen2-Math and Qwen2-Math-Instruct-1.5B/7B/72B. Qwen2-Math is a series of specialized math language models built upon the Qwen2 LLMs, which significantly outperforms the mathematical capabilities of open-source models and even closed-source models (e.g., GPT4o). We hope that Qwen2-Math can contribute to the scientific community for solving advanced mathematical problems that require complex, multi-step logical reasoning.
16
+
17
+
18
+ ## Model Details
19
+
20
+
21
+ For more details, please refer to our [blog post](https://qwenlm.github.io/blog/qwen2-math/) and [GitHub repo](https://github.com/QwenLM/Qwen2-Math).
22
+
23
+
24
+ ## Requirements
25
+ * `transformers>=4.40.0` for Qwen2-Math models. The latest version is recommended.
26
+
27
+ > [!Warning]
28
+ > <div align="center">
29
+ > <b>
30
+ > 🚨 This is a must because `transformers` integrated Qwen2 codes since `4.37.0`.
31
+ > </b>
32
+ > </div>
33
+
34
+ For requirements on GPU memory and the respective throughput, see similar results of Qwen2 [here](https://qwen.readthedocs.io/en/latest/benchmark/speed_benchmark.html).
35
+
36
+ ## Quick Start
37
+
38
+ > [!Important]
39
+ >
40
+ > **Qwen2-Math-7B-Instruct** is an instruction model for chatting;
41
+ >
42
+ > **Qwen2-Math-7B** is a base model typically used for completion and few-shot inference, serving as a better starting point for fine-tuning.
43
+ >
44
+
45
+ ### 🤗 Hugging Face Transformers
46
+
47
+ Qwen2-Math can be deployed and inferred in the same way as [Qwen2](https://github.com/QwenLM/Qwen2). Here we show a code snippet to show you how to use the chat model with `transformers`:
48
+
49
+ ```python
50
+ from transformers import AutoModelForCausalLM, AutoTokenizer
51
+
52
+ model_name = "Qwen/Qwen2-Math-7B-Instruct"
53
+ device = "cuda" # the device to load the model onto
54
+
55
+ model = AutoModelForCausalLM.from_pretrained(
56
+ model_name,
57
+ torch_dtype="auto",
58
+ device_map="auto"
59
+ )
60
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
61
+
62
+ prompt = "Find the value of $x$ that satisfies the equation $4x+5 = 6x+7$."
63
+ messages = [
64
+ {"role": "system", "content": "You are a helpful assistant."},
65
+ {"role": "user", "content": prompt}
66
+ ]
67
+ text = tokenizer.apply_chat_template(
68
+ messages,
69
+ tokenize=False,
70
+ add_generation_prompt=True
71
+ )
72
+ model_inputs = tokenizer([text], return_tensors="pt").to(device)
73
+
74
+ generated_ids = model.generate(
75
+ **model_inputs,
76
+ max_new_tokens=512
77
+ )
78
+ generated_ids = [
79
+ output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
80
+ ]
81
+
82
+ response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
83
+ ```
84
+
85
+ ### 🤖 ModelScope
86
+ We strongly advise users, especially those in mainland China, to use ModelScope. `snapshot_download` can help you solve issues concerning downloading checkpoints.
87
+
88
+
89
+ ## Citation
90
+
91
+ If you find our work helpful, feel free to give us a citation.
92
+
93
+ ```
94
+ @article{yang2024qwen2,
95
+ title={Qwen2 technical report},
96
+ 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},
97
+ journal={arXiv preprint arXiv:2407.10671},
98
+ year={2024}
99
+ }
100
+ ```