Update README.md
Browse files
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=
|
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 |
|