srinivasbilla commited on
Commit
a3ded10
·
verified ·
1 Parent(s): 1cb4955

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -12,11 +12,11 @@ from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, WhisperTokeni
12
  from kokoro import KPipeline
13
  import soundfile as sf
14
  import subprocess
15
- subprocess.run(
16
- "pip install flash-attn --no-build-isolation",
17
- env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
18
- shell=True,
19
- )
20
 
21
  device = "cuda" if torch.cuda.is_available() else "cpu"
22
  torch_dtype = torch.float16
 
12
  from kokoro import KPipeline
13
  import soundfile as sf
14
  import subprocess
15
+ # subprocess.run(
16
+ # "pip install flash-attn --no-build-isolation",
17
+ # env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
18
+ # shell=True,
19
+ # )
20
 
21
  device = "cuda" if torch.cuda.is_available() else "cpu"
22
  torch_dtype = torch.float16