Update configs.yaml

tatsu-lab · Jul 17, 2024 · b8334cd · b8334cd
1 parent a181c98
commit b8334cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/alpaca_eval/models_configs/gemma-2-9b-it-DPO/configs.yaml b/src/alpaca_eval/models_configs/gemma-2-9b-it-DPO/configs.yaml
@@ -5,12 +5,12 @@ gemma-2-9b-it-DPO:
     max_new_tokens: 4096
     model_kwargs:
       torch_dtype: bfloat16
-    model_name: /scratch/gpfs/mengzhou/space17/v2/ae2_submission/gemma-2-9b-it-DPO
+    model_name: princeton-nlp/gemma-2-9b-it-DPO
     stop_token_ids:
     - 1
     - 107
     temperature: 0.5
     top_p: 1.0
   fn_completions: vllm_local_completions
   pretty_name: gemma-2-9b-it-DPO
-  prompt_template: gemma-2-9b-it-DPO/prompt.txt
+  prompt_template: gemma-2-9b-it-DPO/prompt.txt