Update README.md
Browse files
README.md
CHANGED
@@ -12,4 +12,69 @@ tags:
|
|
12 |
- open
|
13 |
- r1
|
14 |
- math
|
15 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
12 |
- open
|
13 |
- r1
|
14 |
- math
|
15 |
+
---
|
16 |
+
# **Open-R1-Math-7B-Instruct**
|
17 |
+
|
18 |
+
The *Open-R1-Math-7B-Instruct* is a fine-tuned language model designed for advanced reasoning and instruction‐following tasks. It leverages the Qwen2.5-7B base model and has been fine-tuned on a chain of thought reasoning dataset derived from [OpenR1-Math-220k](https://huggingface.co/datasets/open-r1/OpenR1-Math-220k). This model is optimized for tasks requiring logical reasoning, detailed explanations, and multi-step problem-solving, making it ideal for applications such as instruction-following, text generation, and complex reasoning tasks.
|
19 |
+
|
20 |
+
# **Quickstart with Transformers**
|
21 |
+
|
22 |
+
Below is a code snippet using `apply_chat_template` to show how to load the tokenizer and model and how to generate content:
|
23 |
+
|
24 |
+
```python
|
25 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
26 |
+
|
27 |
+
model_name = "Open-R1-Math-7B-Instruct"
|
28 |
+
|
29 |
+
model = AutoModelForCausalLM.from_pretrained(
|
30 |
+
model_name,
|
31 |
+
torch_dtype="auto",
|
32 |
+
device_map="auto"
|
33 |
+
)
|
34 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
35 |
+
|
36 |
+
prompt = "How many r in strawberry."
|
37 |
+
messages = [
|
38 |
+
{"role": "system", "content": "You are a helpful and harmless assistant. You are Qwen developed by Alibaba. You should think step-by-step."},
|
39 |
+
{"role": "user", "content": prompt}
|
40 |
+
]
|
41 |
+
text = tokenizer.apply_chat_template(
|
42 |
+
messages,
|
43 |
+
tokenize=False,
|
44 |
+
add_generation_prompt=True
|
45 |
+
)
|
46 |
+
model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
|
47 |
+
|
48 |
+
generated_ids = model.generate(
|
49 |
+
**model_inputs,
|
50 |
+
max_new_tokens=512
|
51 |
+
)
|
52 |
+
generated_ids = [
|
53 |
+
output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
|
54 |
+
]
|
55 |
+
|
56 |
+
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
57 |
+
```
|
58 |
+
|
59 |
+
# **Intended Use**
|
60 |
+
|
61 |
+
The Open-R1-Math-7B-Instruct model is designed for advanced reasoning and instruction-following tasks, with specific applications including:
|
62 |
+
|
63 |
+
1. **Instruction Following**: Providing detailed and step-by-step guidance for a wide range of user queries.
|
64 |
+
2. **Logical Reasoning**: Solving problems requiring multi-step thought processes, such as math problems or complex logic-based scenarios.
|
65 |
+
3. **Text Generation**: Crafting coherent, contextually relevant, and well-structured text in response to prompts.
|
66 |
+
4. **Problem-Solving**: Analyzing and addressing tasks that require chain-of-thought (CoT) reasoning, making it ideal for education, tutoring, and technical support.
|
67 |
+
5. **Knowledge Enhancement**: Leveraging reasoning datasets to offer deeper insights and explanations for a wide variety of topics.
|
68 |
+
|
69 |
+
# **Limitations**
|
70 |
+
|
71 |
+
1. **Data Bias**: As the model is fine-tuned on specific datasets, its outputs may reflect inherent biases from the training data.
|
72 |
+
2. **Context Limitation**: Performance may degrade for tasks requiring knowledge or reasoning that significantly exceeds the model's pretraining or fine-tuning context.
|
73 |
+
3. **Complexity Ceiling**: While optimized for multi-step reasoning, exceedingly complex or abstract problems may result in incomplete or incorrect outputs.
|
74 |
+
4. **Dependency on Prompt Quality**: The quality and specificity of the user prompt heavily influence the model's responses.
|
75 |
+
5. **Non-Factual Outputs**: Despite being fine-tuned for reasoning, the model can still generate hallucinated or factually inaccurate content, particularly for niche or unverified topics.
|
76 |
+
6. **Computational Requirements**: Running the model effectively requires significant computational resources, particularly when generating long sequences or handling high-concurrency workloads.
|
77 |
+
|
78 |
+
---
|
79 |
+
|
80 |
+
This version reflects the new name *Open-R1-Math-7B-Instruct* and specifies that its fine-tuning data comes from the [OpenR1-Math-220k dataset](https://huggingface.co/datasets/open-r1/OpenR1-Math-220k).
|