diff --git a/backend/python/bark/requirements-cpu.txt b/backend/python/bark/requirements-cpu.txt index e64648085..12e376ade 100644 --- a/backend/python/bark/requirements-cpu.txt +++ b/backend/python/bark/requirements-cpu.txt @@ -1,4 +1,4 @@ transformers accelerate -torch==2.7.1+xpu +torch==2.4.1 torchaudio==2.4.1 \ No newline at end of file diff --git a/backend/python/bark/requirements-cublas12.txt b/backend/python/bark/requirements-cublas12.txt index a1ee4ea6e..537169495 100644 --- a/backend/python/bark/requirements-cublas12.txt +++ b/backend/python/bark/requirements-cublas12.txt @@ -1,4 +1,4 @@ -torch==2.7.1+xpu +torch==2.4.1 torchaudio==2.4.1 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 8ac076012..4e1fef6cf 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.4 -torch==2.7.1+xpu +torch==2.8.0+rocm6.4 torchaudio==2.8.0+rocm6.4 transformers accelerate \ No newline at end of file