diff --git a/llm_on_ray/finetune/models/llama-2-7b-chat-hf.yaml b/llm_on_ray/finetune/models/llama-2-7b-chat-hf.yaml index 173869fe7..5d2b5b7fd 100644 --- a/llm_on_ray/finetune/models/llama-2-7b-chat-hf.yaml +++ b/llm_on_ray/finetune/models/llama-2-7b-chat-hf.yaml @@ -1,5 +1,5 @@ General: - base_model: NousResearch/Llama-2-7b-chat-hf + base_model: meta-llama/Llama-2-7b-chat-hf gpt_base_model: false output_dir: /tmp/llm-ray/output save_strategy: no diff --git a/llm_on_ray/finetune/models/llama-7b.yaml b/llm_on_ray/finetune/models/llama-7b.yaml index 8f32ffb63..5c7253b8f 100644 --- a/llm_on_ray/finetune/models/llama-7b.yaml +++ b/llm_on_ray/finetune/models/llama-7b.yaml @@ -1,5 +1,5 @@ General: - base_model: NousResearch/Llama-2-7b-hf + base_model: meta-llama/Llama-2-7b-hf gpt_base_model: false output_dir: /tmp/llm-ray/output save_strategy: no