update2

HabanaAI · Mar 4, 2025 · 42a6207 · 42a6207
1 parent 7d6b24a
commit 42a6207
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.jenkins/lm-eval-harness/test_lm_eval_correctness.py b/.jenkins/lm-eval-harness/test_lm_eval_correctness.py
@@ -79,12 +79,12 @@ def launch_lm_eval(eval_config):
 
     if eval_config.get("num_scheduler_steps"):
         model_args += \
-            f",num_scheduler_steps={eval_config.get('num_scheduler_steps')}"
+            f"num_scheduler_steps={eval_config.get('num_scheduler_steps')}"
 
     # if LORA_ADAPTER_PATH:
     #     model_args += f",peft={LORA_ADAPTER_PATH}"
     if LORA_ADAPTER_PATH:
-        model_args += f",lora_local_path={LORA_ADAPTER_PATH}"
+        model_args += f"enable_lora=True,lora_local_path={LORA_ADAPTER_PATH}"
 
     kwargs = {}
     if 'fewshot_as_multiturn' in eval_config: