Skip to content

Commit 0d1a1ab

Browse files
committed
update
Signed-off-by: minmingzhu <minming.zhu@intel.com>
1 parent 2d16324 commit 0d1a1ab

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

.github/workflows/workflow_finetune.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,7 @@ jobs:
145145
146146
- name: Run Deltatuner Test on DENAS-LoRA Model
147147
run: |
148-
if [[ ${{ matrix.model }} =~ ^(mosaicml\/mpt-7b-chat|huggyllama\/llama-7b|meta-llama\/Llama-2-7b-chat-hf|mistralai\/Mistral-7B-v0.1|google\/gemma-2b)$ ]]; then
148+
if [[ ${{ matrix.model }} =~ ^(mosaicml\/mpt-7b-chat|huggyllama\/llama-7b|meta-llama\/Llama-2-7b-chat-hf|mistralai\/Mistral-7B-v0.1|mistralai\/Mixtral-8x7B-Instruct-v0.1)$ ]]; then
149149
echo ${{ matrix.model }} is not supported!
150150
else
151151
docker exec "finetune" bash -c "rm -rf /tmp/llm-ray/*"

.github/workflows/workflow_inference.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ jobs:
158158
- name: Run Inference Test with DeepSpeed
159159
run: |
160160
TARGET=${{steps.target.outputs.target}}
161-
if [[ ${{ matrix.model }} =~ ^(gpt2|falcon-7b|starcoder|mpt-7b.*)$ ]]; then
161+
if [[ ${{ matrix.model }} =~ ^(mixtral-8x7b-Instruct-v0.1|gpt2|falcon-7b|starcoder|mpt-7b.*)$ ]]; then
162162
echo ${{ matrix.model }} is not supported!
163163
elif [[ ! ${{ matrix.model }} == "llama-2-7b-chat-hf-vllm" ]]; then
164164
docker exec "${TARGET}" bash -c "python .github/workflows/config/update_inference_config.py --config_file llm_on_ray/inference/models/\"${{ matrix.model }}\".yaml --output_file \"${{ matrix.model }}\".yaml.deepspeed --deepspeed"
@@ -171,7 +171,7 @@ jobs:
171171
if: ${{ matrix.dtuner_model }}
172172
run: |
173173
TARGET=${{steps.target.outputs.target}}
174-
if [[ ${{ matrix.model }} =~ ^(gpt2|falcon-7b|starcoder|mpt-7b.*)$ ]]; then
174+
if [[ ${{ matrix.model }} =~ ^(mixtral-8x7b-Instruct-v0.1|gpt2|falcon-7b|starcoder|mpt-7b.*)$ ]]; then
175175
echo ${{ matrix.model }} is not supported!
176176
else
177177
docker exec "${TARGET}" bash -c "llm_on_ray-serve --config_file .github/workflows/config/mpt_deltatuner_deepspeed.yaml --simple"

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ dependencies = [
3939

4040
[project.optional-dependencies]
4141
cpu = [
42-
"transformers>=4.35.0, <=4.35.2",
42+
"transformers>=4.36.0, <=4.38.1",
4343
"intel_extension_for_pytorch>=2.2.0",
4444
"torch>=2.2.0",
4545
"oneccl_bind_pt>=2.2.0"

0 commit comments

Comments
 (0)