Skip to content
This repository was archived by the owner on Sep 23, 2025. It is now read-only.

Commit 0503b6d

Browse files
committed
support mistralai/Mixtral-8x7B-Instruct-v0.1 for CPU
Signed-off-by: minmingzhu <[email protected]>
1 parent 92c5a15 commit 0503b6d

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

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

0 commit comments

Comments
 (0)