Update README.md

README.md

@@ -4,7 +4,7 @@ datasets:
 - nvidia/OpenMathInstruct-1
 language:
 - en
-library_name: transformers
+library_name: nemo
 tags:
 - nvidia
 - code
@@ -14,31 +14,81 @@ tags:
 
 # OpenMath-CodeLlama-7b-Python
 
-## Description:
-
 OpenMath models were designed to solve mathematical problems by integrating text-based reasoning with code blocks
 executed by Python interpreter. The models were trained on [OpenMathInstruct-1](https://huggingface.co/datasets/nvidia/OpenMathInstruct-1),
 a math instruction tuning dataset with 1.8M problem-solution pairs generated using permissively licensed
 [Mixtral-8x7B](https://huggingface.co/mistralai/Mixtral-8x7B-v0.1) model.
 
-
-| Model                                            | Size  | GSM8K     | MATH     |
-|--------------------------------------------------|-------|-----------|----------|
-| GPT-4 [1]                                        |   -   |    94.4   |   56.2   |
-| GPT-4 + code [2]                                 |   -   |    92.9   |   69.7   |
-| OpenMath-CodeLlama-7B ([nemo](https://huggingface.co/nvidia/OpenMath-CodeLlama-7b-Python), [HF](https://huggingface.co/nvidia/OpenMath-CodeLlama-7b-Python-hf))                 |   7B  |    75.9   |   43.6   |
-| OpenMath-CodeLlama-7B + self-consistency (k=50)  |   7B  |    84.8   |   55.6   |
-| OpenMath-Mistral-7B ([nemo](https://huggingface.co/nvidia/OpenMath-Mistral-7B-v0.1), [HF](https://huggingface.co/nvidia/OpenMath-Mistral-7B-v0.1-hf))                              |   7B  |    80.2   |   44.5   |
-| OpenMath-Mistral-7B + self-consistency (k=50)    |   7B  |    86.9   |   57.2   |
-| OpenMath-CodeLlama-13B ([nemo](https://huggingface.co/nvidia/OpenMath-CodeLlama-13b-Python), [HF](https://huggingface.co/nvidia/OpenMath-CodeLlama-13b-Python-hf))                           |  13B  |    78.8   |   45.5   |
-| OpenMath-CodeLlama-13B + self-consistency (k=50) |  13B  |    86.8   |   57.6   |
-| OpenMath-CodeLlama-34B ([nemo](https://huggingface.co/nvidia/OpenMath-CodeLlama-34b-Python), [HF](https://huggingface.co/nvidia/OpenMath-CodeLlama-34b-Python-hf))                           |  34B  |    80.7   |   48.3   |
-| OpenMath-CodeLlama-34B + self-consistency (k=50) |  34B  |    88.0   |   60.2   |
-| OpenMath-Llama2-70B ([nemo](https://huggingface.co/nvidia/OpenMath-Llama-2-70b), [HF](https://huggingface.co/nvidia/OpenMath-Llama-2-70b-hf))                              |  70B  |    84.7   |   46.3   |
-| OpenMath-Llama2-70B + self-consistency (k=50)    |  70B  |    90.1   |   58.3   |
-| OpenMath-CodeLlama-70B ([nemo](https://huggingface.co/nvidia/OpenMath-CodeLlama-70b-Python), [HF](https://huggingface.co/nvidia/OpenMath-CodeLlama-70b-Python-hf))                           |  70B  |  **84.6**      |   **50.7**    |
-| OpenMath-CodeLlama-70B + self-consistency (k=50) |  70B  |  **90.8**    |   **60.4**    |
-
+<table border="1">
+  <tr>
+    <td></td>
+    <td colspan="2" style="text-align: center;">greedy</td>
+    <td colspan="2" style="text-align: center;">majority@50</td>
+  </tr>
+  <tr>
+    <td style="text-align: center;">model</td>
+    <td style="text-align: center;">GSM8K</td>
+    <td style="text-align: center;">MATH</td>
+    <td style="text-align: center;">GMS8K</td>
+    <td style="text-align: center;">MATH</td>
+  </tr>
+  <tr>
+    <td style="text-align: right;">GPT-4 <a href="https://arxiv.org/abs/2312.08935">[1]</a></td>
+    <td style="text-align: center;">94.4</td>
+    <td style="text-align: center;">56.2</td>
+    <td style="text-align: center;">-</td>
+    <td style="text-align: center;">-</td>
+  </tr>
+  <tr>
+    <td style="text-align: right;">GPT-4 + code <a href="https://arxiv.org/abs/2308.07921v1">[2]</a></td>
+    <td style="text-align: center;">92.9</td>
+    <td style="text-align: center;">69.7</td>
+    <td style="text-align: center;">-</td>
+    <td style="text-align: center;">-</td>
+  </tr>
+  <tr>
+    <td style="text-align: right;">OpenMath-CodeLlama-7B (<a href="https://huggingface.co/nvidia/OpenMath-CodeLlama-7b-Python">nemo</a> | <a href="https://huggingface.co/nvidia/OpenMath-CodeLlama-7b-Python-hf">HF</a>)</td>
+    <td style="text-align: center;">75.9</td>
+    <td style="text-align: center;">43.6</td>
+    <td style="text-align: center;">84.8</td>
+    <td style="text-align: center;">55.6</td>
+  </tr>
+  <tr>
+    <td style="text-align: right;">OpenMath-Mistral-7B (<a href="https://huggingface.co/nvidia/OpenMath-Mistral-7B-v0.1">nemo</a> | <a href="https://huggingface.co/nvidia/OpenMath-Mistral-7B-v0.1-hf">HF</a>)</td>
+    <td style="text-align: center;">80.2</td>
+    <td style="text-align: center;">44.5</td>
+    <td style="text-align: center;">86.9</td>
+    <td style="text-align: center;">57.2</td>
+  </tr>
+  <tr>
+    <td style="text-align: right;">OpenMath-CodeLlama-13B (<a href="https://huggingface.co/nvidia/OpenMath-CodeLlama-13b-Python">nemo</a> | <a href="https://huggingface.co/nvidia/OpenMath-CodeLlama-13b-Python-hf">HF</a>)</td>
+    <td style="text-align: center;">78.8</td>
+    <td style="text-align: center;">45.5</td>
+    <td style="text-align: center;">86.8</td>
+    <td style="text-align: center;">57.6</td>
+  </tr>
+  <tr>
+    <td style="text-align: right;">OpenMath-CodeLlama-34B (<a href="https://huggingface.co/nvidia/OpenMath-CodeLlama-34b-Python">nemo</a> | <a href="https://huggingface.co/nvidia/OpenMath-CodeLlama-34b-Python-hf">HF</a>)</td>
+    <td style="text-align: center;">80.7</td>
+    <td style="text-align: center;">48.3</td>
+    <td style="text-align: center;">88.0</td>
+    <td style="text-align: center;">60.2</td>
+  </tr>
+  <tr>
+    <td style="text-align: right;">OpenMath-Llama2-70B (<a href="https://huggingface.co/nvidia/OpenMath-Llama-2-70b">nemo</a> | <a href="https://huggingface.co/nvidia/OpenMath-Llama-2-70b-hf">HF</a>)</td>
+    <td style="text-align: center;"><b>84.7</b></td>
+    <td style="text-align: center;">46.3</td>
+    <td style="text-align: center;">90.1</td>
+    <td style="text-align: center;">58.3</td>
+  </tr>
+  <tr>
+    <td style="text-align: right;">OpenMath-CodeLlama-70B (<a href="https://huggingface.co/nvidia/OpenMath-CodeLlama-70b-Python">nemo</a> | <a href="https://huggingface.co/nvidia/OpenMath-CodeLlama-70b-Python-hf">HF</a>)</td>
+    <td style="text-align: center;">84.6</td>
+    <td style="text-align: center;"><b>50.7</b></td>
+    <td style="text-align: center;"><b>90.8</b></td>
+    <td style="text-align: center;"><b>60.4</b></td>
+  </tr>
+</table>
 
 The pipeline we used to produce these models is fully open-sourced under a commercially permissive license.
 
@@ -69,7 +119,7 @@ Please see [NeMo-Skills Github repo](https://github.com/Kipok/NeMo-Skills) for t
 
 ## Contact
 
-E-Mail: [Igor Gitman](mailto:[email protected])
+E-Mail Igor Gitman at [email protected]
 
 ## Citation
 
@@ -78,4 +128,5 @@ If you find this model useful, please cite the following works
 TODO
 
 ## License
+
 The use of this model is governed by the [Llama 2 Community License Agreement](https://ai.meta.com/llama/license/)