diff --git a/backend/python/bark/requirements-cpu.txt b/backend/python/bark/requirements-cpu.txt index 12e376ad..01e64567 100644 --- a/backend/python/bark/requirements-cpu.txt +++ b/backend/python/bark/requirements-cpu.txt @@ -1,4 +1,4 @@ transformers accelerate torch==2.4.1 -torchaudio==2.4.1 \ No newline at end of file +torchaudio==2.6.0+cu118 \ No newline at end of file diff --git a/backend/python/bark/requirements-cublas11.txt b/backend/python/bark/requirements-cublas11.txt index 9f8fe9ff..e0a95a6c 100644 --- a/backend/python/bark/requirements-cublas11.txt +++ b/backend/python/bark/requirements-cublas11.txt @@ -1,5 +1,5 @@ --extra-index-url https://download.pytorch.org/whl/cu118 torch==2.4.1+cu118 -torchaudio==2.4.1+cu118 +torchaudio==2.6.0+cu118 transformers accelerate \ No newline at end of file diff --git a/backend/python/bark/requirements-cublas12.txt b/backend/python/bark/requirements-cublas12.txt index 53716949..acaf08b9 100644 --- a/backend/python/bark/requirements-cublas12.txt +++ b/backend/python/bark/requirements-cublas12.txt @@ -1,4 +1,4 @@ torch==2.4.1 -torchaudio==2.4.1 +torchaudio==2.6.0+cu118 transformers accelerate \ No newline at end of file diff --git a/backend/python/bark/requirements-hipblas.txt b/backend/python/bark/requirements-hipblas.txt index 1d54fb16..a6eb827f 100644 --- a/backend/python/bark/requirements-hipblas.txt +++ b/backend/python/bark/requirements-hipblas.txt @@ -1,5 +1,5 @@ --extra-index-url https://download.pytorch.org/whl/rocm6.0 torch==2.4.1+rocm6.0 -torchaudio==2.4.1+rocm6.0 +torchaudio==2.6.0+cu118 transformers accelerate \ No newline at end of file diff --git a/backend/python/bark/requirements-intel.txt b/backend/python/bark/requirements-intel.txt index f24bd166..f7f2b79e 100644 --- a/backend/python/bark/requirements-intel.txt +++ b/backend/python/bark/requirements-intel.txt @@ -1,7 +1,7 @@ --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ intel-extension-for-pytorch==2.3.110+xpu torch==2.3.1+cxx11.abi -torchaudio==2.3.1+cxx11.abi +torchaudio==2.6.0+cu118 oneccl_bind_pt==2.3.100+xpu optimum[openvino] setuptools