czd358121692 commited on
Commit
6451b6b
1 Parent(s): a5ab46e

flash-attn

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -14,4 +14,4 @@ accelerate
14
  safetensors
15
  bitsandbytes
16
  scipy
17
- flash-attn
 
14
  safetensors
15
  bitsandbytes
16
  scipy
17
+ flash-attn --index-url https://download.pytorch.org/whl/cu121