mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-10 10:40:44 +08:00
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
"""
|
|
LM eval harness on model to compare vs HF baseline computed offline.
|
|
Configs are found in configs/$MODEL.yaml
|
|
|
|
* export LM_EVAL_TEST_DATA_FILE=configs/Meta-Llama-3-70B-Instruct.yaml
|
|
* export LM_EVAL_TP_SIZE=4
|
|
* pytest -s test_lm_eval_correctness.py
|
|
"""
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
import lm_eval
|
|
import numpy
|
|
import yaml
|
|
|
|
RTOL = 0.02
|
|
TEST_DATA_FILE = os.environ.get(
|
|
"LM_EVAL_TEST_DATA_FILE",
|
|
".buildkite/lm-eval-harness/configs/Meta-Llama-3-8B-Instruct.yaml")
|
|
|
|
TP_SIZE = os.environ.get("LM_EVAL_TP_SIZE", 1)
|
|
|
|
|
|
def launch_lm_eval(eval_config):
|
|
model_args = f"pretrained={eval_config['model_name']}," \
|
|
f"tensor_parallel_size={TP_SIZE}," \
|
|
f"add_bos_token=true"
|
|
|
|
results = lm_eval.simple_evaluate(
|
|
model="vllm",
|
|
model_args=model_args,
|
|
tasks=[task["name"] for task in eval_config["tasks"]],
|
|
num_fewshot=eval_config["num_fewshot"],
|
|
limit=eval_config["limit"],
|
|
batch_size="auto")
|
|
|
|
return results
|
|
|
|
|
|
def test_lm_eval_correctness():
|
|
eval_config = yaml.safe_load(
|
|
Path(TEST_DATA_FILE).read_text(encoding="utf-8"))
|
|
|
|
# Launch eval requests.
|
|
results = launch_lm_eval(eval_config)
|
|
|
|
# Confirm scores match ground truth.
|
|
for task in eval_config["tasks"]:
|
|
for metric in task["metrics"]:
|
|
ground_truth = metric["value"]
|
|
measured_value = results["results"][task["name"]][metric["name"]]
|
|
print(f'{task["name"]} | {metric["name"]}: '
|
|
f'ground_truth={ground_truth} | measured={measured_value}')
|
|
assert numpy.isclose(ground_truth, measured_value, rtol=RTOL)
|