13 lines
394 B
YAML
13 lines
394 B
YAML
# For vllm script, with -t option (tensor parallel size).
|
|
# 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
|
|
model_name: "mgoin/Minitron-4B-Base-FP8"
|
|
tasks:
|
|
- name: "gsm8k"
|
|
metrics:
|
|
- name: "exact_match,strict-match"
|
|
value: 0.231
|
|
- name: "exact_match,flexible-extract"
|
|
value: 0.22
|
|
limit: 1000
|
|
num_fewshot: 5
|