fix(parler-tts): pin torchaudio and torch for hipblas

Signed-off-by: Ettore Di Giacinto <[email protected]>
dave-gray101 · Aug 21, 2024 · 2bb48b0 · 2bb48b0
1 parent 023ce59
commit 2bb48b0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/backend/python/parler-tts/requirements-hipblas.txt b/backend/python/parler-tts/requirements-hipblas.txt
@@ -1,5 +1,5 @@
 --extra-index-url https://download.pytorch.org/whl/rocm6.0
-torch
-torchaudio
+torch==2.3.0+rocm6.0
+torchaudio==2.3.0+rocm6.0
 transformers
-accelerate
+accelerate