Update daily_tests.yaml

add_xcomposer_testcase
zhulinJulia24 2024-07-17 13:06:59 +08:00 committed by GitHub
parent 7122bf0cd3
commit 626f2c096f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions

View File

@ -41,7 +41,8 @@ jobs:
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 lmdeploy[all] beautifulsoup4 decord lxml
pip install sentencepiece auto-gptq==0.6.0 beautifulsoup4 decord lxml
pip install lmdeploy[all] --extra-index-url https://download.pytorch.org/whl/cu118
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