Fix typo

HabanaAI · Oct 30, 2024 · 2df2f1f · 2df2f1f
1 parent 246aab1
commit 2df2f1f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.jenkins/lm-eval-harness/configs/Meta-Llama-3.1-8B-Instruct-mss.yaml b/.jenkins/lm-eval-harness/configs/Meta-Llama-3.1-8B-Instruct-mss.yaml
@@ -13,4 +13,4 @@ num_fewshot: 8
 dtype: "bfloat16"
 fewshot_as_multiturn: true
 apply_chat_template: true
-num_scheduled_steps: 10
+num_scheduler_steps: 10
diff --git a/.jenkins/lm-eval-harness/test_lm_eval_correctness.py b/.jenkins/lm-eval-harness/test_lm_eval_correctness.py
@@ -42,9 +42,9 @@ def launch_lm_eval(eval_config):
                  f"max_model_len=4096," \
                  f"max_num_seqs={max_num_seqs}," \
                  f"trust_remote_code={trust_remote_code}"
-    if eval_config.get("num_scheduled_steps"):
+    if eval_config.get("num_scheduler_steps"):
         model_args += \
-            f",num_scheduled_steps={eval_config.get('num_scheduled_steps')}"
+            f",num_scheduler_steps={eval_config.get('num_scheduler_steps')}"
         print(f"MODEL_ARGS: {model_args}")
     kwargs = {}
     if 'fewshot_as_multiturn' in eval_config: