diff --git a/backend/python/rerankers/requirements-cpu.txt b/backend/python/rerankers/requirements-cpu.txt index 93bae5883925..e27a47263797 100644 --- a/backend/python/rerankers/requirements-cpu.txt +++ b/backend/python/rerankers/requirements-cpu.txt @@ -1,4 +1,4 @@ transformers accelerate -torch==2.8.0 +torch==2.4.1 rerankers[transformers] \ No newline at end of file diff --git a/backend/python/rerankers/requirements-cublas12.txt b/backend/python/rerankers/requirements-cublas12.txt index 93bae5883925..e27a47263797 100644 --- a/backend/python/rerankers/requirements-cublas12.txt +++ b/backend/python/rerankers/requirements-cublas12.txt @@ -1,4 +1,4 @@ transformers accelerate -torch==2.8.0 +torch==2.4.1 rerankers[transformers] \ No newline at end of file diff --git a/backend/python/rerankers/requirements-cublas13.txt b/backend/python/rerankers/requirements-cublas13.txt index c12f4674e52b..b565a9cc154a 100644 --- a/backend/python/rerankers/requirements-cublas13.txt +++ b/backend/python/rerankers/requirements-cublas13.txt @@ -1,5 +1,5 @@ --extra-index-url https://download.pytorch.org/whl/cu130 transformers accelerate -torch==2.8.0 +torch==2.9.1 rerankers[transformers] \ No newline at end of file diff --git a/backend/python/rerankers/requirements-hipblas.txt b/backend/python/rerankers/requirements-hipblas.txt index 1701ffe29ed2..7a72b3d0650f 100644 --- a/backend/python/rerankers/requirements-hipblas.txt +++ b/backend/python/rerankers/requirements-hipblas.txt @@ -1,5 +1,5 @@ --extra-index-url https://download.pytorch.org/whl/rocm6.4 transformers accelerate -torch==2.8.0 +torch==2.8.0+rocm6.4 rerankers[transformers] \ No newline at end of file diff --git a/backend/python/rerankers/requirements-intel.txt b/backend/python/rerankers/requirements-intel.txt index 8704ab189c48..820dd84224a7 100644 --- a/backend/python/rerankers/requirements-intel.txt +++ b/backend/python/rerankers/requirements-intel.txt @@ -2,7 +2,7 @@ intel-extension-for-pytorch==2.3.110+xpu transformers accelerate -torch==2.8.0 +torch==2.3.1+cxx11.abi oneccl_bind_pt==2.8.0+xpu rerankers[transformers] optimum[openvino]