github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-11 21:01:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CI] use flash_attn==2.7.4.post1

Browse Source
This commit is contained in:
botbw 2025-07-10 12:42:42 +08:00
parent 4b7727996e
commit 12c55538f5
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/build_on_pr.yml vendored
View File

@ -138,6 +138,10 @@ jobs:
cp -p -r /github/home/cuda_ext_cache/* /__w/ColossalAI/ColossalAI/ cp -p -r /github/home/cuda_ext_cache/* /__w/ColossalAI/ColossalAI/
fi fi
- name: Install flash-attention
run: |
pip install flash-attn==2.7.4.post1 --no-build-isolation
- name: Install Colossal-AI - name: Install Colossal-AI
run: | run: |
BUILD_EXT=1 pip install -v -e . BUILD_EXT=1 pip install -v -e .