Add text-generation pipeline tag
Browse filesThis PR adds the `text-generation` pipeline tag to the model card. This ensures the model can be found via the pipeline tag at https://huggingface.co/models?pipeline_tag=text-generation
README.md
CHANGED
@@ -1,15 +1,16 @@
|
|
1 |
---
|
2 |
-
license: llama3.1
|
3 |
base_model:
|
4 |
- meta-llama/Llama-3.1-70B
|
5 |
datasets:
|
6 |
- nvidia/OpenMathInstruct-2
|
7 |
language:
|
8 |
- en
|
|
|
|
|
9 |
tags:
|
10 |
- nvidia
|
11 |
- math
|
12 |
-
|
13 |
---
|
14 |
|
15 |
# OpenMath2-Llama3.1-70B
|
@@ -23,10 +24,10 @@ The model outperforms [Llama3.1-70B-Instruct](https://huggingface.co/meta-llama/
|
|
23 |
| Model | GSM8K | MATH | AMC 2023 | AIME 2024 | Omni-MATH |
|
24 |
|:---|:---:|:---:|:---:|:---:|:---:|
|
25 |
| Llama3.1-8B-Instruct | 84.5 | 51.9 | 9/40 | 2/30 | 12.7 |
|
26 |
-
| OpenMath2-Llama3.1-8B ([nemo](https://huggingface.co/nvidia/OpenMath2-Llama3.1-8B-nemo)
|
27 |
| + majority@256 | 94.1 | 76.1 | 23/40 | 3/30 | 24.6 |
|
28 |
| Llama3.1-70B-Instruct | 95.8 | 67.9 | 19/40 | 6/30 | 19.0 |
|
29 |
-
| **OpenMath2-Llama3.1-70B** ([nemo](https://huggingface.co/nvidia/OpenMath2-Llama3.1-70B-nemo)
|
30 |
| + majority@256 | 96.0 | 79.6 | 24/40 | 6/30 | 27.6 |
|
31 |
|
32 |
The pipeline we used to produce the data and models is fully open-sourced!
|
@@ -61,7 +62,9 @@ pipeline = transformers.pipeline(
|
|
61 |
messages = [
|
62 |
{
|
63 |
"role": "user",
|
64 |
-
"content": "Solve the following math problem. Make sure to put the answer (and only answer) inside \\boxed{}
|
|
|
|
|
65 |
"What is the minimum value of $a^2+6a-7$?"},
|
66 |
]
|
67 |
|
|
|
1 |
---
|
|
|
2 |
base_model:
|
3 |
- meta-llama/Llama-3.1-70B
|
4 |
datasets:
|
5 |
- nvidia/OpenMathInstruct-2
|
6 |
language:
|
7 |
- en
|
8 |
+
library_name: transformers
|
9 |
+
license: llama3.1
|
10 |
tags:
|
11 |
- nvidia
|
12 |
- math
|
13 |
+
pipeline_tag: text-generation
|
14 |
---
|
15 |
|
16 |
# OpenMath2-Llama3.1-70B
|
|
|
24 |
| Model | GSM8K | MATH | AMC 2023 | AIME 2024 | Omni-MATH |
|
25 |
|:---|:---:|:---:|:---:|:---:|:---:|
|
26 |
| Llama3.1-8B-Instruct | 84.5 | 51.9 | 9/40 | 2/30 | 12.7 |
|
27 |
+
| OpenMath2-Llama3.1-8B ([nemo](https://huggingface.co/nvidia/OpenMath2-Llama3.1-8B-nemo) | [HF](https://huggingface.co/nvidia/OpenMath2-Llama3.1-8B)) | 91.7 | 67.8 | 16/40 | 3/30 | 22.0 |
|
28 |
| + majority@256 | 94.1 | 76.1 | 23/40 | 3/30 | 24.6 |
|
29 |
| Llama3.1-70B-Instruct | 95.8 | 67.9 | 19/40 | 6/30 | 19.0 |
|
30 |
+
| **OpenMath2-Llama3.1-70B** ([nemo](https://huggingface.co/nvidia/OpenMath2-Llama3.1-70B-nemo) | [HF](https://huggingface.co/nvidia/OpenMath2-Llama3.1-70B)) | 94.9 | 71.9 | 20/40 | 4/30 | 23.1 |
|
31 |
| + majority@256 | 96.0 | 79.6 | 24/40 | 6/30 | 27.6 |
|
32 |
|
33 |
The pipeline we used to produce the data and models is fully open-sourced!
|
|
|
62 |
messages = [
|
63 |
{
|
64 |
"role": "user",
|
65 |
+
"content": "Solve the following math problem. Make sure to put the answer (and only answer) inside \\boxed{}.
|
66 |
+
|
67 |
+
" +
|
68 |
"What is the minimum value of $a^2+6a-7$?"},
|
69 |
]
|
70 |
|