integrate llama-2-7b-hf (intel#173)

Signed-off-by: minmingzhu <[email protected]>
yutianchen666 · Apr 7, 2024 · 4bb33d3 · 4bb33d3
1 parent e70fd9a
commit 4bb33d3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/workflow_finetune.yml b/.github/workflows/workflow_finetune.yml
@@ -34,7 +34,7 @@ jobs:
     name: finetune
     strategy:
       matrix:
-        model: [ EleutherAI/gpt-j-6b, meta-llama/Llama-2-7b-chat-hf, gpt2, bigscience/bloom-560m, facebook/opt-125m, mosaicml/mpt-7b-chat, huggyllama/llama-7b, mistralai/Mistral-7B-v0.1, google/gemma-2b]
+        model: [ EleutherAI/gpt-j-6b, meta-llama/Llama-2-7b-chat-hf, gpt2, bigscience/bloom-560m, facebook/opt-125m, mosaicml/mpt-7b-chat, meta-llama/Llama-2-7b-hf, mistralai/Mistral-7B-v0.1, google/gemma-2b]
         isPR:
           - ${{inputs.ci_type == 'pr'}}
 

diff --git 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: meta-llama/Llama-2-7b
+  base_model: meta-llama/Llama-2-7b-hf
   gpt_base_model: false
   output_dir: /tmp/llm-ray/output
   checkpoint_dir: /tmp/llm-ray/checkpoint