diff --git a/backend/python/vall-e-x/requirements-hipblas.txt b/backend/python/vall-e-x/requirements-hipblas.txt index 6ddd0b8d2971..fc43790a2e59 100644 --- 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 \ No newline at end of file +torch==2.3.0+rocm6.0 +torchaudio==2.3.0+rocm6.0 \ No newline at end of file