File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -34,13 +34,12 @@ jobs:
3434 unit_test :
3535 runs-on : [self-hosted, linux-a100-s2]
3636 timeout-minutes : 4320 # 72hours
37- env :
38- HF_TOKEN : ${{ secrets.HF_TOKEN }}
3937 container :
4038 image : nvcr.io/nvidia/tritonserver:22.12-py3
4139 options : " --gpus=all --ipc=host --user root -e PIP_CACHE_DIR=/root/.cache/pip -e CUDA_VISIBLE_DEVICES=2,3 --pull never"
4240 volumes :
4341 - /nvme/share_data/github-actions/pip-cache:/root/.cache/pip
42+ - /nvme/share_data/github-actions/hf_home:/root/.cache/huggingface
4443 - /nvme/share_data/github-actions/packages:/root/packages
4544 - /usr/share/zoneinfo/Asia/Shanghai:/etc/localtime:ro
4645 steps :
8281 python3 -m pip install /root/packages/flash_attn-2.6.3+cu118torch2.3cxx11abiFALSE-cp38-cp38-linux_x86_64.whl
8382 python3 -m pip install -r requirements_cuda.txt -r requirements/test.txt
8483 python3 -m pip install .
84+ huggingface-cli login --token ${{ secrets.HF_TOKEN }}
85+ huggingface-cli whoami
86+ pytest tests/test_lmdeploy/test_auto_backend.py
8587 - name : Check env
8688 run : |
8789 python3 -m pip list
You can’t perform that action at this time.
0 commit comments