dongyh20 commited on
Commit
0088ddf
·
1 Parent(s): f59c9d1

update space

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -25,7 +25,7 @@ import librosa
25
  import whisper
26
 
27
  import torchaudio
28
-
29
  def install_cuda_toolkit():
30
  # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
31
  CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
@@ -44,8 +44,6 @@ def install_cuda_toolkit():
44
  os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
45
 
46
  install_cuda_toolkit()
47
-
48
- import subprocess
49
  subprocess.run('pip install flash-attn==2.5.9.post1 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "True"}, shell=True)
50
 
51
  import sys
 
25
  import whisper
26
 
27
  import torchaudio
28
+ import subprocess
29
  def install_cuda_toolkit():
30
  # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
31
  CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
 
44
  os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
45
 
46
  install_cuda_toolkit()
 
 
47
  subprocess.run('pip install flash-attn==2.5.9.post1 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "True"}, shell=True)
48
 
49
  import sys