From 89cabd477c686b648861252c244356fbeaaf6f27 Mon Sep 17 00:00:00 2001
From: zhulinJulia24 <145004780+zhulinJulia24@users.noreply.github.com>
Date: Thu, 18 Jul 2024 17:54:03 +0800
Subject: [PATCH] Update daily_tests.yaml

---
 .github/workflows/daily_tests.yaml | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/daily_tests.yaml b/.github/workflows/daily_tests.yaml
index e94be3d..4e6c539 100644
--- a/.github/workflows/daily_tests.yaml
+++ b/.github/workflows/daily_tests.yaml
@@ -36,12 +36,17 @@ jobs:
         conda create -n internlm-model-latest --clone ${CONDA_BASE_ENV}
         source activate internlm-model-latest
         pip install transformers
+    - name: install torch
+      run: |
+        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]
+        python3 -m pip list
+        conda deactivate
     - name: run_test
       run: |
         source activate internlm-model-latest
-        pip install torch==2.2.2 torchvision==0.17.2 --index-url https://download.pytorch.org/whl/cu118
-        pip install /mnt/petrelfs/qa-caif-cicd/resource/flash_attn-2.5.8+cu118torch2.2cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
-        pip install sentencepiece auto-gptq==0.6.0 beautifulsoup4 decord lxml
         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