From 06b841e133ca920ddfab0a1fdb9db6c4952d282c Mon Sep 17 00:00:00 2001 From: Junya Morioka Date: Sun, 27 Oct 2024 01:13:24 +0900 Subject: [PATCH] Update build.yml --- .github/workflows/build.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index f165b42..ab93930 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -135,6 +135,7 @@ jobs: 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=1 + cd flash-attention FLASH_ATTENTION_FORCE_BUILD="TRUE" python setup.py bdist_wheel --dist-dir=dist wheel_name=$(basename $(ls dist/*.whl | head -n 1)) echo "wheel_name=$wheel_name" >> $GITHUB_ENV