Add pipeline tag: text-generation
Browse filesThis PR adds the `pipeline_tag: text-generation` to the model card metadata. This will improve discoverability of the model on the Hugging Face model hub. The model is used for mathematical reasoning, which falls under text generation.
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/).
|