From bda47d41aaf484ac8cb4f97b975135801ca9bc38 Mon Sep 17 00:00:00 2001 From: zhulin1 Date: Thu, 29 Feb 2024 11:38:24 +0800 Subject: [PATCH] daily ci add lmdeploy install --- .github/workflows/daily_tests.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/daily_tests.yaml b/.github/workflows/daily_tests.yaml index 1a8285f..e27522d 100644 --- a/.github/workflows/daily_tests.yaml +++ b/.github/workflows/daily_tests.yaml @@ -27,7 +27,7 @@ jobs: conda create -n internlm-model-latest --clone ${CONDA_BASE_ENV} source activate internlm-model-latest pip install transformers==${{ matrix.transformers-version }} - pip install sentencepiece auto-gptq + pip install sentencepiece auto-gptq lmdeploy[all] srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=${GITHUB_RUN_ID}-${GITHUB_JOB} --gpus-per-task=2 pytest -s -v --color=yes ./tests/test_hf_model.py conda deactivate - name: load_latest_hf_model @@ -36,7 +36,7 @@ jobs: conda create -n internlm-model-latest --clone ${CONDA_BASE_ENV} source activate internlm-model-latest pip install transformers - pip install sentencepiece auto-gptq + pip install sentencepiece auto-gptq lmdeploy[all] srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=${GITHUB_RUN_ID}-${GITHUB_JOB} --gpus-per-task=2 pytest -s -v --color=yes ./tests/test_hf_model.py conda deactivate - name: remove_env