fix(vall-e-x): pin hipblas deps

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
mudler · Aug 8, 2024 · 8b85687 · 8b85687
1 parent 60117ec
commit 8b85687
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/python/vall-e-x/requirements-hipblas.txt b/backend/python/vall-e-x/requirements-hipblas.txt
@@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/rocm6.0
 accelerate
-torch
-torchaudio
+torch==2.3.0+rocm6.0
+torchaudio==2.3.0+rocm6.0