| 1 | # For vllm script, with -t option (tensor parallel size). |
| 2 | # bash .buildkite/lm-eval-harness/run-lm-eval-gsm-vllm-baseline.sh -m mgoin/Minitron-4B-Base-FP8 -b auto -l 1000 -f 5 -t 1 |
| 3 | model_name: "mgoin/Minitron-4B-Base-FP8" |
| 4 | tasks: |
| 5 | - name: "gsm8k" |
| 6 | metrics: |
| 7 | - name: "exact_match,strict-match" |
| 8 | value: 0.231 |
| 9 | - name: "exact_match,flexible-extract" |
| 10 | value: 0.22 |
| 11 | limit: 1000 |
| 12 | num_fewshot: 5 |