YUNTA88 commited on
Commit
0aad208
·
verified ·
1 Parent(s): 4c88a9b

Upload root_scripts/install_flash_attn.sh with huggingface_hub

Browse files
Files changed (1) hide show
  1. root_scripts/install_flash_attn.sh +17 -0
root_scripts/install_flash_attn.sh ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+ set -x
3
+
4
+ echo "=== Trying flash-attn install ==="
5
+
6
+ # Method 1: Try FA3 from the official wheel index
7
+ echo "--- Method 1: FA3 wheel index ---"
8
+ pip install flash-attn \
9
+ --extra-index-url https://dao-ailab.github.io/flash-attention/wheels/ \
10
+ --no-build-isolation \
11
+ 2>&1 | tail -20
12
+
13
+ # Check if it worked
14
+ python3 -c "import flash_attn; print(f'flash_attn version: {flash_attn.__version__}')" 2>&1
15
+
16
+ echo "INSTALL_SCRIPT_DONE"
17
+