From a9ac204e099d645703397f599b0006e1c2708b74 Mon Sep 17 00:00:00 2001 From: ljvmiranda921 Date: Mon, 27 Jan 2025 16:41:28 -0800 Subject: [PATCH] Add attn_implementation default value for experiments --- experiments/run_llm_evals.sh | 1 + 1 file changed, 1 insertion(+) diff --git a/experiments/run_llm_evals.sh b/experiments/run_llm_evals.sh index ca20f4c..e1c6816 100755 --- a/experiments/run_llm_evals.sh +++ b/experiments/run_llm_evals.sh @@ -71,6 +71,7 @@ for lang_code in "${!languages[@]}"; do --split "filtered" \ --output_dir "$OUTDIR" \ --include_languages "${languages[$lang_code]}" "English" \ + --attn_implementation "flash_attention_2" \ --trust_remote_code \ --force_local done