Skip to content

Commit a57e718

Browse files
authored
fix ci
1 parent 8d82842 commit a57e718

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

.github/workflows/run_chatgpt_examples.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,8 @@ jobs:
3131

3232
- name: Install flash-attention
3333
run: |
34+
pip uninstall flash-attn
35+
pip install torch==2.5.1
3436
pip install flash-attn==2.7.4.post1 --no-build-isolation
3537
3638
- name: Install Colossal-AI

0 commit comments

Comments
 (0)