Skip to content

Commit 6e4dd84

Browse files
authored
fix ci; specify flash-attn version
1 parent 605bf88 commit 6e4dd84

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

.github/workflows/run_chatgpt_examples.yml

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

32-
- name: Install Colossal-AI
33-
run: |
34-
pip install --no-cache-dir -v -e .
35-
3632
- name: Install flash-attention
3733
run: |
38-
pip install https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
39-
# https://release-assets.githubusercontent.com/github-production-release-asset/494232964/eea5ccc0-9cb8-48d3-ac9d-049198a38582?sp=r&sv=2018-11-09&sr=b&spr=https&se=2025-11-11T11%3A24%3A12Z&rscd=attachment%3B+filename%3Dflash_attn-2.7.4.post1%2Bcu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl&rsct=application%2Foctet-stream&skoid=96c2d410-5711-43a1-aedd-ab1947aa7ab0&sktid=398a6654-997b-47e9-b12b-9515b896b4de&skt=2025-11-11T10%3A23%3A28Z&ske=2025-11-11T11%3A24%3A12Z&sks=b&skv=2018-11-09&sig=PBlHLl81QvE2jqPG%2FFFmmwW5rUBiKKZCiponNJIUnoU%3D&jwt=eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmVsZWFzZS1hc3NldHMuZ2l0aHVidXNlcmNvbnRlbnQuY29tIiwia2V5Ijoia2V5MSIsImV4cCI6MTc2Mjg2MDc1OCwibmJmIjoxNzYyODU3MTU4LCJwYXRoIjoicmVsZWFzZWFzc2V0cHJvZHVjdGlvbi5ibG9iLmNvcmUud2luZG93cy5uZXQifQ.ZB3u9Nzw0xCKvUCbpTXXku8fPsNRcq65pMztucY0N9g&response-content-disposition=attachment%3B%20filename%3Dflash_attn-2.7.4.post1%2Bcu12torch2.5cxx11abiFALSE-cp310-cp310-linux_x86_64.whl&response-content-type=application%2Foctet-stream
34+
pip install flash-attn==2.7.4.post1 --no-build-isolation
35+
36+
- name: Install Colossal-AI
37+
run: |
38+
BUILD_EXT=1 pip install --no-cache-dir -v -e .
39+
4040
- name: Install ChatGPT
4141
env:
4242
CFLAGS: "-O1"
@@ -47,9 +47,9 @@ jobs:
4747
pip install --no-cache-dir -v .
4848
pip install --no-cache-dir -r examples/requirements.txt
4949
50-
# - name: Install Transformers
51-
# run: |
52-
# pip install --no-cache-dir transformers==4.36.2
50+
- name: Install Transformers
51+
run: |
52+
pip install --no-cache-dir transformers==4.36.2
5353
5454
- name: Execute Examples
5555
run: |

0 commit comments

Comments
 (0)