yindo/flash-attention-prebuild-wheels-rocm
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git synced 2026-06-30 23:57:53 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

Update build.yml

Browse Source
This commit is contained in:
Junya Morioka
2024-10-27 03:53:58 +09:00
committed by GitHub
parent 06b841e133
commit 38dd0a6eff
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build.yml
+3 -3
View File
@@ -81,7 +81,7 @@ jobs:
- name: Set Swap Space
uses: pierotofy/set-swap-space@master
with:
swap-size-gb: 10
swap-size-gb: 48
- uses: actions/setup-python@v5
with:
@@ -134,7 +134,7 @@ jobs:
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=1
export MAX_JOBS=2
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))
@@ -142,7 +142,7 @@ jobs:
- name: Install Test
run: |
pip install flash_attention/dist/${{ env.wheel_name }}
pip install flash-attention/dist/${{ env.wheel_name }}
python -c "import flash_atten; print(flash_atten.__version__)"
- name: Get the tag version