Skip to content

Commit 74c3310

Browse files
committed
support mistralai/Mixtral-8x7B-Instruct-v0.1 for CPU
Signed-off-by: minmingzhu <minming.zhu@intel.com>
1 parent 82e242c commit 74c3310

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

.github/workflows/workflow_finetune.yml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ jobs:
3434
name: finetune
3535
strategy:
3636
matrix:
37-
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 ]
37+
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, mistralai/Mixtral-8x7B-Instruct-v0.1 ]
3838
isPR:
3939
- ${{inputs.ci_type == 'pr'}}
4040

@@ -44,6 +44,7 @@ jobs:
4444
- { model: "EleutherAI/gpt-j-6b"}
4545
- { model: "meta-llama/Llama-2-7b-chat-hf"}
4646
- { model: "mistralai/Mistral-7B-v0.1"}
47+
- { model: "mistralai/Mixtral-8x7B-Instruct-v0.1"}
4748

4849
runs-on: self-hosted
4950

@@ -131,6 +132,8 @@ jobs:
131132
}
132133
if "${{ matrix.model }}" == "mistralai/Mistral-7B-v0.1":
133134
result['General']['lora_config']['target_modules'] = ["q_proj", "k_proj", "v_proj", "o_proj", "gate_proj", "up_proj", "down_proj", "lm_head",]
135+
elif "${{ matrix.model }}" == "mistralai/Mixtral-8x7B-Instruct-v0.1":
136+
result['General']['lora_config']['target_modules'] = ["k_proj", "v_proj"]
134137
else:
135138
result['General']['lora_config']['target_modules'] = None
136139
with open(conf_path, 'w') as output:

0 commit comments

Comments
 (0)