Skip to content

Commit d3c87a4

Browse files
committed
fix ci; specify flash-attn version
1 parent c494856 commit d3c87a4

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

.github/workflows/run_chatgpt_examples.yml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,10 +29,6 @@ jobs:
2929
- name: Checkout ColossalAI
3030
uses: actions/checkout@v2
3131

32-
- name: Install flash-attention
33-
run: |
34-
pip install flash-attn==2.7.4.post1 --no-build-isolation
35-
3632
- name: Install Colossal-AI
3733
run: |
3834
pip install --no-cache-dir -v -e .
@@ -47,6 +43,10 @@ jobs:
4743
pip install --no-cache-dir -v .
4844
pip install --no-cache-dir -r examples/requirements.txt
4945
46+
- name: Install flash-attention
47+
run: |
48+
pip install flash-attn==2.7.4.post1 --no-build-isolation
49+
5050
- name: Install Transformers
5151
run: |
5252
pip install --no-cache-dir transformers==4.36.2

0 commit comments

Comments
 (0)