diff --git a/backend/python/transformers-musicgen/requirements-cpu.txt b/backend/python/transformers-musicgen/requirements-cpu.txt index 2021fc20..5d2e3845 100644 --- a/backend/python/transformers-musicgen/requirements-cpu.txt +++ b/backend/python/transformers-musicgen/requirements-cpu.txt @@ -1,3 +1,3 @@ transformers accelerate -torch==2.4.1 \ No newline at end of file +torch==2.5.1+cxx11.abi \ No newline at end of file diff --git a/backend/python/transformers-musicgen/requirements-cublas11.txt b/backend/python/transformers-musicgen/requirements-cublas11.txt index cd2c9fdb..c1641b9b 100644 --- a/backend/python/transformers-musicgen/requirements-cublas11.txt +++ b/backend/python/transformers-musicgen/requirements-cublas11.txt @@ -1,4 +1,4 @@ --extra-index-url https://download.pytorch.org/whl/cu118 transformers accelerate -torch==2.4.1+cu118 \ No newline at end of file +torch==2.5.1+cxx11.abi \ No newline at end of file diff --git a/backend/python/transformers-musicgen/requirements-cublas12.txt b/backend/python/transformers-musicgen/requirements-cublas12.txt index 2021fc20..5d2e3845 100644 --- a/backend/python/transformers-musicgen/requirements-cublas12.txt +++ b/backend/python/transformers-musicgen/requirements-cublas12.txt @@ -1,3 +1,3 @@ transformers accelerate -torch==2.4.1 \ No newline at end of file +torch==2.5.1+cxx11.abi \ No newline at end of file diff --git a/backend/python/transformers-musicgen/requirements-hipblas.txt b/backend/python/transformers-musicgen/requirements-hipblas.txt index 122b2032..6834b9ce 100644 --- a/backend/python/transformers-musicgen/requirements-hipblas.txt +++ b/backend/python/transformers-musicgen/requirements-hipblas.txt @@ -1,4 +1,4 @@ --extra-index-url https://download.pytorch.org/whl/rocm6.0 transformers accelerate -torch==2.4.1+rocm6.0 \ No newline at end of file +torch==2.5.1+cxx11.abi \ No newline at end of file