From d2b21ce95a99133aee0969fa4b45bf1d0393a501 Mon Sep 17 00:00:00 2001 From: xiamengzhou Date: Wed, 17 Jul 2024 13:53:19 -0400 Subject: [PATCH] Update configs.yaml --- .../models_configs/gemma-2-9b-it-SimPO/configs.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/alpaca_eval/models_configs/gemma-2-9b-it-SimPO/configs.yaml b/src/alpaca_eval/models_configs/gemma-2-9b-it-SimPO/configs.yaml index 7f46cbf7..3400aef5 100644 --- a/src/alpaca_eval/models_configs/gemma-2-9b-it-SimPO/configs.yaml +++ b/src/alpaca_eval/models_configs/gemma-2-9b-it-SimPO/configs.yaml @@ -5,7 +5,7 @@ gemma-2-9b-it-SimPO: max_new_tokens: 4096 model_kwargs: torch_dtype: bfloat16 - model_name: /scratch/gpfs/mengzhou/space17/v2/ae2_submission/gemma-2-9b-it-SimPO + model_name: princeton-nlp/gemma-2-9b-it-SimPO stop_token_ids: - 1 - 107 @@ -13,4 +13,4 @@ gemma-2-9b-it-SimPO: top_p: 1.0 fn_completions: vllm_local_completions pretty_name: gemma-2-9b-it-SimPO - prompt_template: gemma-2-9b-it-DPO/prompt.txt \ No newline at end of file + prompt_template: gemma-2-9b-it-DPO/prompt.txt