From 2bb48b08162f14938079d1b0c30b0032037412a6 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Wed, 21 Aug 2024 18:27:20 +0200 Subject: [PATCH] fix(parler-tts): pin torchaudio and torch for hipblas Signed-off-by: Ettore Di Giacinto --- backend/python/parler-tts/requirements-hipblas.txt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/python/parler-tts/requirements-hipblas.txt b/backend/python/parler-tts/requirements-hipblas.txt index af9e820e9241..b8758537835e 100644 --- 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 \ No newline at end of file +accelerate