jiang-cc commited on
Commit
17e9caa
·
verified ·
1 Parent(s): 87fe738

fix: use correct flash-attn wheel for torch2.9+cu126+cp310 from mjun0812

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -2,7 +2,7 @@ torch
2
  torchvision
3
  transformers==4.57.3
4
  accelerate
5
- flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3%2Bcu12torch2.8cxx11abiTRUE-cp310-cp310-linux_x86_64.whl
6
  qwen_vl_utils
7
  Pillow
8
  spaces
 
2
  torchvision
3
  transformers==4.57.3
4
  accelerate
5
+ flash-attn @ https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.0/flash_attn-2.8.3%2Bcu126torch2.9-cp310-cp310-linux_x86_64.whl
6
  qwen_vl_utils
7
  Pillow
8
  spaces