fix: deps issue with cuda

janhq · Dec 18, 2024 · a4f9181 · a4f9181
1 parent e21a6d0
commit a4f9181
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/models/fish-speech/requirements.cuda.txt b/models/fish-speech/requirements.cuda.txt
@@ -20,8 +20,6 @@ vector_quantize_pytorch==1.14.24
 resampy>=0.4.3
 einx[torch]==0.2.2
 zstandard>=0.22.0
-pydub
-pyaudio
 faster_whisper
 modelscope==1.17.1
 funasr==1.1.5
@@ -34,5 +32,5 @@ cachetools
 fastapi
 
 # Stable dependencies
-torch<=2.4.1
+torch<=2.4.1 --index-url https://download.pytorch.org/whl/cu121
 torchaudio
diff --git a/models/fish-speech/requirements.txt b/models/fish-speech/requirements.txt
@@ -20,8 +20,6 @@ vector_quantize_pytorch==1.14.24
 resampy>=0.4.3
 einx[torch]==0.2.2
 zstandard>=0.22.0
-pydub
-pyaudio
 faster_whisper
 modelscope==1.17.1
 funasr==1.1.5