Fix conflicts + YAML

Xilinx · Dec 20, 2024 · 52dcfe0 · 52dcfe0
1 parent f7e1f48
commit 52dcfe0
Showing 1 changed file with 9 additions and 7 deletions.
diff --git 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