reduce number of models for finetuning (intel#130)

Signed-off-by: jiafu zhang <[email protected]>
JoshuaL3000 · Nov 21, 2023 · 734031a · 734031a
1 parent 54625aa
commit 734031a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/night_build_memo.txt b/.github/workflows/night_build_memo.txt
@@ -0,0 +1 @@
+finetune: gpt2, bigscience/bloom-560m, facebook/opt-125m, mosaicml/mpt-7b-chat, huggyllama/llama-7b
diff --git a/.github/workflows/workflow_finetune.yml b/.github/workflows/workflow_finetune.yml
@@ -12,7 +12,7 @@ jobs:
     name: finetune test
     strategy:
       matrix:
-        model: [ EleutherAI/gpt-j-6b, gpt2, bigscience/bloom-560m, facebook/opt-125m, mosaicml/mpt-7b-chat, huggyllama/llama-7b, meta-llama/Llama-2-7b-chat-hf ]
+        model: [ EleutherAI/gpt-j-6b, meta-llama/Llama-2-7b-chat-hf ]
     runs-on: self-hosted
     steps:
       - name: Checkout
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		finetune: gpt2, bigscience/bloom-560m, facebook/opt-125m, mosaicml/mpt-7b-chat, huggyllama/llama-7b