yuki-imajuku commited on
Commit
f79b053
·
verified ·
1 Parent(s): 1518cc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  # Install FlashAttention
2
  import subprocess
3
  subprocess.run(
4
- "pip install flash-attn --no-build-isolation",
5
  env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
6
  shell=True,
7
  )
 
1
  # Install FlashAttention
2
  import subprocess
3
  subprocess.run(
4
+ "pip install flash-attn==2.7.4.post1 --no-build-isolation",
5
  env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
6
  shell=True,
7
  )