tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

use chat template

Browse Source
This commit is contained in:
Wing Lian
2025-01-11 23:18:27 -05:00
parent 5c226b600d
commit bfc91a91ca
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/axolotl/integrations/lm_eval/cli.py
View File

@@ -53,6 +53,7 @@ def build_lm_eval_command(
str(batch_size), str(batch_size),
"--output_path", "--output_path",
output_path, output_path,
"--apply_chat_template",
] ]
wandb_args = [] wandb_args = []
if wandb_project: if wandb_project:
@@ -65,6 +66,7 @@ def build_lm_eval_command(
if num_fewshot_val: if num_fewshot_val:
lm_eval_args.append("--num_fewshot") lm_eval_args.append("--num_fewshot")
lm_eval_args.append(str(num_fewshot_val)) lm_eval_args.append(str(num_fewshot_val))
# lm_eval_args.append("--fewshot_as_multiturn")
yield lm_eval_args yield lm_eval_args