Skip to content

Commit b42b2cb

Browse files
authored
fix ci
1 parent a57e718 commit b42b2cb

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

.github/workflows/run_chatgpt_examples.yml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ jobs:
3232
- name: Install flash-attention
3333
run: |
3434
pip uninstall flash-attn
35-
pip install torch==2.5.1
35+
pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu121
3636
pip install flash-attn==2.7.4.post1 --no-build-isolation
3737
3838
- name: Install Colossal-AI
@@ -46,12 +46,12 @@ jobs:
4646
MAX_JOBS: 4
4747
run: |
4848
cd applications/ColossalChat
49-
pip install --no-cache-dir -v .
49+
pip install --no-cache-dir -v -e .
5050
pip install --no-cache-dir -r examples/requirements.txt
5151
52-
- name: Install Transformers
53-
run: |
54-
pip install --no-cache-dir transformers==4.36.2
52+
# - name: Install Transformers
53+
# run: |
54+
# pip install --no-cache-dir transformers==4.36.2
5555

5656
- name: Execute Examples
5757
run: |

0 commit comments

Comments
 (0)