Skip to content

GitLab

Unverified Commit 6a77a3d3 authored by Jiaqi Liao's avatar Jiaqi Liao Committed by GitHub
Browse files

Merge pull request #892 from SkqLiao/main 

fix flash_attn whl path in install job
parents b4ad815e a8d15977
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
.github/workflows/install.yml
View file @ 6a77a3d3
...@@ -40,7 +40,8 @@ jobs: ...@@ -40,7 +40,8 @@ jobs:
conda activate ktransformers-dev conda activate ktransformers-dev
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126 pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
pip3 install packaging ninja cpufeature numpy pip3 install packaging ninja cpufeature numpy
pip install ~/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiTRUE-cp311-cp311-linux_x86_64.whl wget
pip install https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.2cxx11abiTRUE-cp311-cp311-linux_x86_64.whl
- name: Install KTransformers - name: Install KTransformers
run: | run: |
source /home/qujing3/anaconda3/etc/profile.d/conda.sh source /home/qujing3/anaconda3/etc/profile.d/conda.sh
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment