diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 949906e26..3bcf6d29c 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -158,8 +158,12 @@ jobs: run: | python -m pip install --upgrade pip python -m pip install wheel pytest pytest-cov nvidia-pyindex - python -m pip install torch==1.10.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html - python -m pip install git+https://github.com/HazyResearch/flash-attention.git + { + python -m pip install torch==1.10.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html + python -m pip install git+https://github.com/HazyResearch/flash-attention.git + } || { + echo "flash attention was not installed." + } pip install -e "client/[test]" pip install -e "server/[tensorrt]" - name: Test