diff --git a/.github/workflows/daily_tests.yaml b/.github/workflows/daily_tests.yaml index 0c0ab32..ff3570a 100644 --- a/.github/workflows/daily_tests.yaml +++ b/.github/workflows/daily_tests.yaml @@ -41,7 +41,10 @@ jobs: source activate internlm-model-latest pip install /mnt/petrelfs/qa-caif-cicd/resource/flash_attn-2.5.8+cu118torch2.2cxx11abiFALSE-cp310-cp310-linux_x86_64.whl pip install torch==2.2.2 torchvision==0.17.2 --index-url https://download.pytorch.org/whl/cu118 - pip install sentencepiece auto-gptq==0.6.0 beautifulsoup4 decord lxml lmdeploy[all]==0.5.0 + pip install sentencepiece auto-gptq==0.6.0 beautifulsoup4 decord lxml + export LMDEPLOY_VERSION=0.5.0 + export PYTHON_VERSION=310 + pip install https://github.com/InternLM/lmdeploy/releases/download/v${LMDEPLOY_VERSION}/lmdeploy-${LMDEPLOY_VERSION}+cu118-cp${PYTHON_VERSION}-cp${PYTHON_VERSION}-manylinux2014_x86_64.whl --extra-index-url https://download.pytorch.org/whl/cu118 python3 -m pip list conda deactivate - name: run_test