diff --git a/backend/python/vall-e-x/requirements-cpu.txt b/backend/python/vall-e-x/requirements-cpu.txt index 0aad881298e..c48986f619c 100644 --- a/backend/python/vall-e-x/requirements-cpu.txt +++ b/backend/python/vall-e-x/requirements-cpu.txt @@ -1,3 +1,3 @@ accelerate torch==2.4.1 -torchaudio==2.4.1 \ No newline at end of file +torchaudio==2.5.1+cu118 \ No newline at end of file diff --git a/backend/python/vall-e-x/requirements-cublas11.txt b/backend/python/vall-e-x/requirements-cublas11.txt index c45de5b7801..ab967c22f2f 100644 --- a/backend/python/vall-e-x/requirements-cublas11.txt +++ b/backend/python/vall-e-x/requirements-cublas11.txt @@ -1,4 +1,4 @@ --extra-index-url https://download.pytorch.org/whl/cu118 accelerate torch==2.4.1+cu118 -torchaudio==2.4.1+cu118 \ No newline at end of file +torchaudio==2.5.1+cu118 \ No newline at end of file diff --git a/backend/python/vall-e-x/requirements-cublas12.txt b/backend/python/vall-e-x/requirements-cublas12.txt index 0aad881298e..c48986f619c 100644 --- a/backend/python/vall-e-x/requirements-cublas12.txt +++ b/backend/python/vall-e-x/requirements-cublas12.txt @@ -1,3 +1,3 @@ accelerate torch==2.4.1 -torchaudio==2.4.1 \ No newline at end of file +torchaudio==2.5.1+cu118 \ No newline at end of file diff --git a/backend/python/vall-e-x/requirements-hipblas.txt b/backend/python/vall-e-x/requirements-hipblas.txt index fc43790a2e5..ff629cc11d9 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==2.3.0+rocm6.0 -torchaudio==2.3.0+rocm6.0 \ No newline at end of file +torchaudio==2.5.1+cu118 \ No newline at end of file