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).
|