diff --git a/backend/python/rerankers/requirements-cpu.txt b/backend/python/rerankers/requirements-cpu.txt index 096833d5085b..c84356876ba7 100644 --- a/backend/python/rerankers/requirements-cpu.txt +++ b/backend/python/rerankers/requirements-cpu.txt @@ -1,5 +1,5 @@ # https://github.com/huggingface/transformers/issues/38269#issuecomment-2900214777 transformers==4.52.1 accelerate -torch==2.4.1 +torch==2.6.0+cu118 rerankers[transformers] \ No newline at end of file diff --git a/backend/python/rerankers/requirements-cublas11.txt b/backend/python/rerankers/requirements-cublas11.txt index 9c42ef5e66e9..e5aa09227609 100644 --- a/backend/python/rerankers/requirements-cublas11.txt +++ b/backend/python/rerankers/requirements-cublas11.txt @@ -2,5 +2,5 @@ # https://github.com/huggingface/transformers/issues/38269#issuecomment-2900214777 transformers==4.52.1 accelerate -torch==2.4.1+cu118 +torch==2.6.0+cu118 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 096833d5085b..c84356876ba7 100644 --- a/backend/python/rerankers/requirements-cublas12.txt +++ b/backend/python/rerankers/requirements-cublas12.txt @@ -1,5 +1,5 @@ # https://github.com/huggingface/transformers/issues/38269#issuecomment-2900214777 transformers==4.52.1 accelerate -torch==2.4.1 +torch==2.6.0+cu118 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 6e796356d762..a4bf321a9d46 100644 --- a/backend/python/rerankers/requirements-hipblas.txt +++ b/backend/python/rerankers/requirements-hipblas.txt @@ -2,5 +2,5 @@ # https://github.com/huggingface/transformers/issues/38269#issuecomment-2900214777 transformers==4.52.1 accelerate -torch==2.4.1+rocm6.0 +torch==2.6.0+cu118 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 5f10a97fd853..587a56617aab 100644 --- a/backend/python/rerankers/requirements-intel.txt +++ b/backend/python/rerankers/requirements-intel.txt @@ -3,7 +3,7 @@ intel-extension-for-pytorch==2.3.110+xpu # https://github.com/huggingface/transformers/issues/38269#issuecomment-2900214777 transformers==4.52.1 accelerate -torch==2.3.1+cxx11.abi +torch==2.6.0+cu118 oneccl_bind_pt==2.3.100+xpu rerankers[transformers] optimum[openvino]