diff --git a/src/brevitas_examples/llm/config/default_template.yml b/src/brevitas_examples/llm/config/default_template.yml index 50f85340b..ef20184ac 100644 --- a/src/brevitas_examples/llm/config/default_template.yml +++ b/src/brevitas_examples/llm/config/default_template.yml @@ -8,6 +8,15 @@ dataset: wikitext2 eval: false export_prefix: null export_target: null +few_shot_compile: false +few_shot_eval: false +few_shot_limit: null +few_shot_tasks: +- arc_challenge +- arc_easy +- winogrande +- piqa +few_shot_zeroshot: false fuse_sequences: false gpfq: false gptq: false @@ -60,10 +69,3 @@ weight_quant_format: int weight_quant_granularity: per_group weight_quant_type: sym weight_scale_precision: float_scale -zero_shot_compile: false -zero_shot_eval: false -zero_shot_tasks: -- arc_challenge -- arc_easy -- winogrande -- piqa