alexnasa commited on
Commit
6f57021
·
verified ·
1 Parent(s): 6a03d8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ try:
32
  flash_attention_wheel = hf_hub_download(
33
  repo_id="alexnasa/flash-attn-3",
34
  repo_type="model",
35
- filename="torch29+cu128/flash_attn_3-3.0.0b1-cp39-abi3-linux_x86_64.whl",
36
  )
37
 
38
  sh(f"pip install {flash_attention_wheel}")
 
32
  flash_attention_wheel = hf_hub_download(
33
  repo_id="alexnasa/flash-attn-3",
34
  repo_type="model",
35
+ filename="128/flash_attn_3-3.0.0b1-cp39-abi3-linux_x86_64.whl",
36
  )
37
 
38
  sh(f"pip install {flash_attention_wheel}")