Skip to content

Commit 6c40069

Browse files
committed
fix
Signed-off-by: Icey <1790571317@qq.com>
1 parent 2e417d3 commit 6c40069

File tree

3 files changed

+5
-2
lines changed

3 files changed

+5
-2
lines changed

tests/e2e/singlecard/models/configs/Qwen3-30B-A3B.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,4 +10,5 @@ num_fewshot: 5
1010
gpu_memory_utilization: 0.6
1111
enable_expert_parallel: True
1212
tensor_parallel_size: 2
13+
apply_chat_template: False
1314
fewshot_as_multiturn: False

tests/e2e/singlecard/models/report_template.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
```bash
1414
export MODEL_ARGS={{ model_args }}
1515
lm_eval --model {{ model_type }} --model_args $MODEL_ARGS --tasks {{ datasets }} \
16-
--apply_chat_template --fewshot_as_multiturn {% if num_fewshot is defined and num_fewshot != "N/A" %} --num_fewshot {{ num_fewshot }} {% endif %} \
16+
--apply_chat_template {{ apply_chat_template }} --fewshot_as_multiturn {{ fewshot_as_multiturn }} {% if num_fewshot is defined and num_fewshot != "N/A" %} --num_fewshot {{ num_fewshot }} {% endif %} \
1717
--limit {{ limit }} --batch_size {{ batch_size}}
1818
```
1919

tests/e2e/singlecard/models/test_lm_eval_correctness.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,8 @@ def generate_report(tp_size, eval_config, report_data, report_template,
7474
model_args=f"'{','.join(f'{k}={v}' for k, v in model_args.items())}'",
7575
model_type=eval_config.get("model", "vllm"),
7676
datasets=",".join([task["name"] for task in eval_config["tasks"]]),
77+
apply_chat_template=eval_config["apply_chat_template"],
78+
fewshot_as_multiturn=eval_config["fewshot_as_multiturn"],
7779
limit=eval_config.get("limit", None),
7880
batch_size="auto",
7981
num_fewshot=eval_config.get("num_fewshot", "N/A"),
@@ -97,7 +99,7 @@ def test_lm_eval_correctness_param(config_filename, tp_size, report_template,
9799
"limit": eval_config.get("limit", None),
98100
"batch_size": "auto",
99101
}
100-
for s in ["num_fewshot", "fewshot_as_multiturn"]:
102+
for s in ["num_fewshot", "fewshot_as_multiturn", "apply_chat_template"]:
101103
val = eval_config.get(s, None)
102104
if val is not None:
103105
eval_params[s] = val

0 commit comments

Comments
 (0)