igitman nielsr HF Staff commited on
Commit
f4a3c0e
·
verified ·
1 Parent(s): 2384570

Add pipeline tag: text-generation (#2)

Browse files

- Add pipeline tag: text-generation (51e7d23afa20cbc5ff2c36e1045897dba3b45b11)


Co-authored-by: Niels Rogge <[email protected]>

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/).