Update README.md

README.md

@@ -131,7 +131,7 @@ model.save_pretrained("Meta-Llama-3.1-70B-Instruct-quantized.w8a8")
 
 The model was evaluated on MMLU, ARC-Challenge, GSM-8K, Hellaswag, Winogrande and TruthfulQA.
 Evaluation was conducted using the Neural Magic fork of [lm-evaluation-harness](https://github.com/neuralmagic/lm-evaluation-harness/tree/llama_3.1_instruct) (branch llama_3.1_instruct) and the [vLLM](https://docs.vllm.ai/en/stable/) engine.
-This version of the lm-evaluation-harness includes versions of ARC-Challenge and GSM-8K that match the prompting style of [Meta-Llama-3.1-Instruct-evals](https://huggingface.co/datasets/meta-llama/Meta-Llama-3.1-70B-Instruct-evals).
+This version of the lm-evaluation-harness includes versions of MMLU, ARC-Challenge and GSM-8K that match the prompting style of [Meta-Llama-3.1-Instruct-evals](https://huggingface.co/datasets/meta-llama/Meta-Llama-3.1-70B-Instruct-evals).
 
 ### Accuracy
 
@@ -142,7 +142,7 @@ This version of the lm-evaluation-harness includes versions of ARC-Challenge and
    </td>
    <td><strong>Meta-Llama-3.1-70B-Instruct </strong>
    </td>
-   <td><strong>Meta-Llama-3.1-70B-Instruct-quantized.w8a8 (this model)</strong>
+   <td><strong>Meta-Llama-3.1-70B-Instruct-quantized.w8a16 (this model)</strong>
    </td>
    <td><strong>Recovery</strong>
    </td>
@@ -150,9 +150,19 @@ This version of the lm-evaluation-harness includes versions of ARC-Challenge and
   <tr>
    <td>MMLU (5-shot)
    </td>
-   <td>82.21
+   <td>83.88
    </td>
-   <td>81.88
+   <td>83.65
+   </td>
+   <td>99.7%
+   </td>
+  </tr>
+  <tr>
+   <td>MMLU (CoT, 0-shot)
+   </td>
+   <td>85.74
+   </td>
+   <td>85.41
    </td>
    <td>99.6%
    </td>
@@ -160,11 +170,11 @@ This version of the lm-evaluation-harness includes versions of ARC-Challenge and
   <tr>
    <td>ARC Challenge (0-shot)
    </td>
-   <td>95.05
+   <td>93.26
    </td>
-   <td>94.97
+   <td>93.26
    </td>
-   <td>99.9%
+   <td>100.0%
    </td>
   </tr>
   <tr>
@@ -210,11 +220,11 @@ This version of the lm-evaluation-harness includes versions of ARC-Challenge and
   <tr>
    <td><strong>Average</strong>
    </td>
-   <td><strong>83.60</strong>
+   <td><strong>83.89</strong>
    </td>
-   <td><strong>83.71</strong>
+   <td><strong>83.96</strong>
    </td>
-   <td><strong>100.1%</strong>
+   <td><strong>100.2%</strong>
    </td>
   </tr>
 </table>
@@ -227,17 +237,30 @@ The results were obtained using the following commands:
 ```
 lm_eval \
   --model vllm \
-  --model_args pretrained="neuralmagic/Meta-Llama-3.1-70B-Instruct-quantized.w8a8",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=4 \
-  --tasks mmlu \
+  --model_args pretrained="neuralmagic/Meta-Llama-3.1-70B-Instruct-quantized.w8a8",dtype=auto,add_bos_token=True,max_model_len=3850,max_gen_toks=10,tensor_parallel_size=1 \
+  --tasks mmlu_llama_3.1_instruct \
+  --fewshot_as_multiturn \
+  --apply_chat_template \
   --num_fewshot 5 \
   --batch_size auto
 ```
 
+#### MMLU-CoT
+```
+lm_eval \
+  --model vllm \
+  --model_args pretrained="neuralmagic/Meta-Llama-3.1-70B-Instruct-quantized.w8a8",dtype=auto,add_bos_token=True,max_model_len=4064,max_gen_toks=1024,tensor_parallel_size=1 \
+  --tasks mmlu_cot_0shot_llama_3.1_instruct \
+  --apply_chat_template \
+  --num_fewshot 0 \
+  --batch_size auto
+```
+
 #### ARC-Challenge
 ```
 lm_eval \
   --model vllm \
-  --model_args pretrained="neuralmagic/Meta-Llama-3.1-70B-Instruct-quantized.w8a8",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=4 \
+  --model_args pretrained="neuralmagic/Meta-Llama-3.1-70B-Instruct-quantized.w8a8",dtype=auto,add_bos_token=True,max_model_len=3940,max_gen_toks=100,tensor_parallel_size=1 \
   --tasks arc_challenge_llama_3.1_instruct \
   --apply_chat_template \
   --num_fewshot 0 \
@@ -248,7 +271,7 @@ lm_eval \
 ```
 lm_eval \
   --model vllm \
-  --model_args pretrained="neuralmagic/Meta-Llama-3.1-70B-Instruct-quantized.w8a8",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=4 \
+  --model_args pretrained="neuralmagic/Meta-Llama-3.1-70B-Instruct-quantized.w8a8",dtype=auto,add_bos_token=True,max_model_len=4096,max_gen_toks=1024,tensor_parallel_size=1 \
   --tasks gsm8k_cot_llama_3.1_instruct \
   --fewshot_as_multiturn \
   --apply_chat_template \
@@ -260,7 +283,7 @@ lm_eval \
 ```
 lm_eval \
   --model vllm \
-  --model_args pretrained="neuralmagic/Meta-Llama-3.1-70B-Instruct-quantized.w8a8",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=4 \
+  --model_args pretrained="neuralmagic/Meta-Llama-3.1-70B-Instruct-quantized.w8a8",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=1 \
   --tasks hellaswag \
   --num_fewshot 10 \
   --batch_size auto
@@ -270,7 +293,7 @@ lm_eval \
 ```
 lm_eval \
   --model vllm \
-  --model_args pretrained="neuralmagic/Meta-Llama-3.1-70B-Instruct-quantized.w8a8",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=4 \
+  --model_args pretrained="neuralmagic/Meta-Llama-3.1-70B-Instruct-quantized.w8a8",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=1 \
   --tasks winogrande \
   --num_fewshot 5 \
   --batch_size auto
@@ -280,7 +303,7 @@ lm_eval \
 ```
 lm_eval \
   --model vllm \
-  --model_args pretrained="neuralmagic/Meta-Llama-3.1-70B-Instruct-quantized.w8a8",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=4 \
+  --model_args pretrained="neuralmagic/Meta-Llama-3.1-70B-Instruct-quantized.w8a8",dtype=auto,add_bos_token=True,max_model_len=4096,tensor_parallel_size=1 \
   --tasks truthfulqa \
   --num_fewshot 0 \
   --batch_size auto