nielsr HF Staff commited on
Commit
d49fe9b
·
verified ·
1 Parent(s): 3f093ed

Add text-generation pipeline tag

Browse files

This 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

Files changed (1) hide show
  1. README.md +8 -5
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
- library_name: transformers
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) \| [HF](https://huggingface.co/nvidia/OpenMath2-Llama3.1-8B)) | 91.7 | 67.8 | 16/40 | 3/30 | 22.0 |
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) \| [HF](https://huggingface.co/nvidia/OpenMath2-Llama3.1-70B)) | 94.9 | 71.9 | 20/40 | 4/30 | 23.1 |
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{}.\n\n" +
 
 
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