Mingsmilet commited on
Commit
8e36fac
·
verified ·
1 Parent(s): b3c5e7f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -24,7 +24,7 @@ from transformers import pipeline
24
  generator = pipeline("text-generation", model="Mingsmilet/Qwen2.5-1.5B-Open-R1-Distill", device="cuda")
25
 
26
  question = "The fraction\n\\[\\frac{\\left(3^{2008}\\right)^2-\\left(3^{2006}\\right)^2}{\\left(3^{2007}\\right)^2-\\left(3^{2005}\\right)^2}\\]\nsimplifies to which of the following?\n$\\mathrm{(A)}\\ 1\\qquad\\mathrm{(B)}\\ \\frac{9}{4}\\qquad\\mathrm{(C)}\\ 3\\qquad\\mathrm{(D)}\\ \\frac{9}{2}\\qquad\\mathrm{(E)}\\ 9$"
27
- output = generator([{"role": "user", "content": question}], max_new_tokens=2048, return_full_text=False)[0]
28
  print(output["generated_text"])
29
  ```
30
 
 
24
  generator = pipeline("text-generation", model="Mingsmilet/Qwen2.5-1.5B-Open-R1-Distill", device="cuda")
25
 
26
  question = "The fraction\n\\[\\frac{\\left(3^{2008}\\right)^2-\\left(3^{2006}\\right)^2}{\\left(3^{2007}\\right)^2-\\left(3^{2005}\\right)^2}\\]\nsimplifies to which of the following?\n$\\mathrm{(A)}\\ 1\\qquad\\mathrm{(B)}\\ \\frac{9}{4}\\qquad\\mathrm{(C)}\\ 3\\qquad\\mathrm{(D)}\\ \\frac{9}{2}\\qquad\\mathrm{(E)}\\ 9$"
27
+ output = generator([{"role": "user", "content": question}], max_new_tokens=5000, return_full_text=False)[0]
28
  print(output["generated_text"])
29
  ```
30