update space

app.py

@@ -26,7 +26,7 @@ import whisper
 
 import torchaudio
 import subprocess
-subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "True"}, shell=True)
+subprocess.run('pip install flash-attn==2.5.9.post1 --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "True"}, shell=True)
 
 import sys
 sys.path.append('./ola/CosyVoice/')

requirements.txt

@@ -16,6 +16,8 @@ openai-whisper
 deepspeed==0.12.2
 loguru
 decord
+torch==2.1.1
+torchvision==0.16.1
 torchaudio
 av
 librosa