soiz1 commited on
Commit
5e93b28
·
verified ·
1 Parent(s): 123c6d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, WhisperTokeni
11
  import subprocess
12
  subprocess.run(
13
  "pip install flash-attn --no-build-isolation",
14
- env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
15
  shell=True,
16
  )
17
 
 
11
  import subprocess
12
  subprocess.run(
13
  "pip install flash-attn --no-build-isolation",
14
+ env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "FALSE"},
15
  shell=True,
16
  )
17