File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed
Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change 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 : |
You can’t perform that action at this time.
0 commit comments