Update daily-run-test.yml

open-compass · Sep 23, 2024 · bd5d582 · bd5d582
1 parent 4bf2284
commit bd5d582
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/.github/workflows/daily-run-test.yml b/.github/workflows/daily-run-test.yml
@@ -67,7 +67,7 @@ jobs:
           . /cpfs01/shared/public/qa-llm-cicd/miniconda3/bin/activate
           conda create -y --name ${{env.CONDA_ENV}}_${{ matrix.cuda_env }} python=3.10
           conda activate ${{env.CONDA_ENV}}_${{ matrix.cuda_env }}
-          pip install opencompass*.whl
+          pip install opencompass*.whl --cache-dir ${{env.PIP_CACHE_PATH}}
           pip install /cpfs01/user/qa-llm-cicd/packages/lmdeploy-0.6.0+cu118-cp310-cp310-manylinux2014_x86_64.whl --cache-dir ${{env.PIP_CACHE_PATH}}
           pip install /cpfs01/user/qa-llm-cicd/packages/vllm-0.6.1.post1+cu118-cp310-cp310-manylinux1_x86_64.whl --cache-dir ${{env.PIP_CACHE_PATH}}
           pip install human_eval transformers protobuf pytest gguf msgspec librosa vllm_flash_attn bitsandbytes --cache-dir ${{env.PIP_CACHE_PATH}}
@@ -83,9 +83,9 @@ jobs:
           . /cpfs01/shared/public/qa-llm-cicd/miniconda3/bin/activate
           conda create -y --name ${{env.CONDA_ENV}}_${{ matrix.cuda_env }} python=3.10
           conda activate ${{env.CONDA_ENV}}_${{ matrix.cuda_env }}
-          pip install opencompass*.whl
-          pip install opencompass[lmdeploy]
-          pip install opencompass[vllm]
+          pip install opencompass*.whl --cache-dir ${{env.PIP_CACHE_PATH}}
+          pip install opencompass[lmdeploy] --cache-dir ${{env.PIP_CACHE_PATH}}
+          pip install opencompass[vllm] --cache-dir ${{env.PIP_CACHE_PATH}}
           pip install human_eval transformers protobuf pytest gguf msgspec librosa vllm_flash_attn bitsandbytes modelscope --cache-dir ${{env.PIP_CACHE_PATH}}
           pip uninstall torch torchvision torchaudio -y
           pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --cache-dir ${{env.PIP_CACHE_PATH}}