back-prop commited on
Commit
2972ae6
·
verified ·
1 Parent(s): 79477d5

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +52 -0
README.md ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: Qwen/Qwen2.5-3B
3
+ license: apache-2.0
4
+ datasets:
5
+ - math
6
+ metrics:
7
+ - accuracy
8
+ pipeline_tag: text-generation
9
+ language:
10
+ - en
11
+ ---
12
+
13
+ # Qwen2.5-3B-GRPO-MATH-1EPOCH
14
+
15
+ **Description:**
16
+ A GRPO-fine-tuned version of Qwen2.5-3B-Instruct trained on the MATH dataset. It is optimized to produce more accurate contest-style math solutions.
17
+
18
+ ---
19
+
20
+ ## Usage
21
+
22
+ ```python
23
+ from transformers import pipeline
24
+
25
+ generator = pipeline(
26
+ "text-generation",
27
+ model="USERNAME/Qwen-2.5-3B-GRPO-Math",
28
+ device="cuda"
29
+ )
30
+
31
+ prompt = "Evaluate the integral ∫₀¹ x² dx."
32
+ result = generator(
33
+ [{"role": "user", "content": prompt}],
34
+ max_new_tokens=50,
35
+ return_full_text=False
36
+ )[0]
37
+ print(result["generated_text"])
38
+ ````
39
+
40
+ ---
41
+
42
+ ## Citation
43
+
44
+ ```bibtex
45
+ @article{sha2024deepseekmath,
46
+ title = {DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models},
47
+ author = {Shao, Zhihong and Wang, Peiyi and Zhu, Qihao and Xu, Runxin and Song, Junxiao and Bi, Xiao and … Guo, Daya},
48
+ journal = {arXiv preprint arXiv:2402.03300},
49
+ year = {2024},
50
+ }
51
+ ```
52
+