Skip to content

Commit

Permalink
support mistralai/Mixtral-8x7B-Instruct-v0.1 for CPU
Browse files Browse the repository at this point in the history
Signed-off-by: minmingzhu <[email protected]>
  • Loading branch information
minmingzhu committed Mar 15, 2024
1 parent 82e242c commit 74c3310
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion .github/workflows/workflow_finetune.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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 ]
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 ]
isPR:
- ${{inputs.ci_type == 'pr'}}

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

runs-on: self-hosted

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

0 comments on commit 74c3310

Please sign in to comment.