Create README.md
Browse files
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-1.5B-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-1.5B-Instruct** is an instruction model for chatting;
|
| 41 |
+
>
|
| 42 |
+
> **Qwen2-Math-1.5B** 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-1.5B-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 |
+
```
|