Skip to content

Commit 0eefac0

Browse files
committed
fix ci; specify flash-attn version
1 parent 90e6e81 commit 0eefac0

1 file changed

Lines changed: 1 addition & 5 deletions

File tree

.github/workflows/run_chatgpt_examples.yml

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,13 +29,9 @@ jobs:
2929
- name: Checkout ColossalAI
3030
uses: actions/checkout@v2
3131

32-
- name: Install PyTorch
33-
run: |
34-
pip install torch==2.5.1
35-
3632
- name: Install flash-attention
3733
run: |
38-
pip install flash-attn==2.7.4.post1 --no-build-isolation
34+
pip install flash-attn==2.5.8 --no-build-isolation
3935
4036
- name: Install Colossal-AI
4137
run: |

0 commit comments

Comments
 (0)