Spaces:
Running
on
Zero
Running
on
Zero
dongyh20
commited on
Commit
·
0088ddf
1
Parent(s):
f59c9d1
update space
Browse files
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
|