Spaces:
Running on Zero
Running on Zero
fix: use correct flash-attn wheel for torch2.9+cu126+cp310 from mjun0812
Browse files- 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/
|
| 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
|