diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 25fe43c..0d31ea9 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -31,7 +31,7 @@ jobs: name: Build wheels and Upload needs: create_releases runs-on: ubuntu-22.04 - timeout-minutes: 6000 + timeout-minutes: 700 strategy: fail-fast: false matrix: @@ -137,12 +137,12 @@ jobs: ${{ env.CACHE_KEY }}- - name: Build wheels - timeout-minutes: 6000 + timeout-minutes: 600 run: | pip install setuptools==68.0.0 ninja packaging wheel export PATH=/usr/local/nvidia/bin:/usr/local/nvidia/lib64:$PATH export LD_LIBRARY_PATH=/usr/local/nvidia/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH - export MAX_JOBS=$(($(nproc) - 1)) + export MAX_JOBS=$(nproc) cd flash-attention FLASH_ATTENTION_FORCE_BUILD="TRUE" python setup.py bdist_wheel --dist-dir=dist base_wheel_name=$(basename $(ls dist/*.whl | head -n 1))