Add pipeline tag: text-generation (#2)
Browse files- Add pipeline tag: text-generation (51e7d23afa20cbc5ff2c36e1045897dba3b45b11)
Co-authored-by: Niels Rogge <[email protected]>
README.md
CHANGED
|
@@ -1,15 +1,16 @@
|
|
| 1 |
---
|
| 2 |
-
license: cc-by-4.0
|
| 3 |
base_model:
|
| 4 |
- Qwen/Qwen2.5-Math-1.5B
|
| 5 |
datasets:
|
| 6 |
- nvidia/OpenMathReasoning
|
| 7 |
language:
|
| 8 |
- en
|
|
|
|
|
|
|
| 9 |
tags:
|
| 10 |
- nvidia
|
| 11 |
- math
|
| 12 |
-
|
| 13 |
---
|
| 14 |
|
| 15 |
# OpenMath-Nemotron-1.5B
|
|
@@ -86,7 +87,9 @@ pipeline = transformers.pipeline(
|
|
| 86 |
messages = [
|
| 87 |
{
|
| 88 |
"role": "user",
|
| 89 |
-
"content": "Solve the following math problem. Make sure to put the answer (and only answer) inside \\boxed{}
|
|
|
|
|
|
|
| 90 |
"What is the minimum value of $a^2+6a-7$?"},
|
| 91 |
]
|
| 92 |
|
|
@@ -215,4 +218,4 @@ NVIDIA believes Trustworthy AI is a shared responsibility and we have establishe
|
|
| 215 |
|
| 216 |
For more detailed information on ethical considerations for this model, please see the Model Card++ [Explainability](./EXPLAINABILITY.md), [Bias](./BIAS.md), [Safety & Security](./SAFETY.md), and [Privacy](./PRIVACY.md) Subcards.
|
| 217 |
|
| 218 |
-
Please report security vulnerabilities or NVIDIA AI Concerns [here](https://www.nvidia.com/en-us/support/submit-security-vulnerability/).
|
|
|
|
| 1 |
---
|
|
|
|
| 2 |
base_model:
|
| 3 |
- Qwen/Qwen2.5-Math-1.5B
|
| 4 |
datasets:
|
| 5 |
- nvidia/OpenMathReasoning
|
| 6 |
language:
|
| 7 |
- en
|
| 8 |
+
library_name: transformers
|
| 9 |
+
license: cc-by-4.0
|
| 10 |
tags:
|
| 11 |
- nvidia
|
| 12 |
- math
|
| 13 |
+
pipeline_tag: text-generation
|
| 14 |
---
|
| 15 |
|
| 16 |
# OpenMath-Nemotron-1.5B
|
|
|
|
| 87 |
messages = [
|
| 88 |
{
|
| 89 |
"role": "user",
|
| 90 |
+
"content": "Solve the following math problem. Make sure to put the answer (and only answer) inside \\boxed{}.
|
| 91 |
+
|
| 92 |
+
" +
|
| 93 |
"What is the minimum value of $a^2+6a-7$?"},
|
| 94 |
]
|
| 95 |
|
|
|
|
| 218 |
|
| 219 |
For more detailed information on ethical considerations for this model, please see the Model Card++ [Explainability](./EXPLAINABILITY.md), [Bias](./BIAS.md), [Safety & Security](./SAFETY.md), and [Privacy](./PRIVACY.md) Subcards.
|
| 220 |
|
| 221 |
+
Please report security vulnerabilities or NVIDIA AI Concerns [here](https://www.nvidia.com/en-us/support/submit-security-vulnerability/).
|