diff --git a/.github/workflows/daily_tests.yaml b/.github/workflows/daily_tests.yaml index c4d6515..e4f7751 100644 --- a/.github/workflows/daily_tests.yaml +++ b/.github/workflows/daily_tests.yaml @@ -41,7 +41,7 @@ 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 + pip install sentencepiece auto-gptq==0.6.0 lmdeploy[all] 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 -m tmp --color=yes ./tests/test_hf_model.py conda deactivate - name: remove_env diff --git a/tests/test_hf_model.py b/tests/test_hf_model.py index 0ef2126..0b25119 100644 --- a/tests/test_hf_model.py +++ b/tests/test_hf_model.py @@ -394,7 +394,7 @@ class TestXcomposer2d5Model: model.tokenizer = tokenizer query = 'Generate the HTML code of this web image with Tailwind CSS.' - image = ['./examples/screenshot.jpg'] + image = ['/mnt/petrelfs/qa-caif-cicd/github_runner/examples/screenshot.jpg'] with torch.autocast(device_type='cuda', dtype=torch.float16): response = model.resume_2_webpage(query, image,