nielsr HF Staff commited on
Commit
51e7d23
·
verified ·
1 Parent(s): 2384570

Add pipeline tag: text-generation

Browse files

This 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.

Files changed (1) hide show
  1. README.md +7 -4
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
- library_name: transformers
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{}.\n\n" +
 
 
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/).