diff --git a/backend/python/rerankers/requirements-cpu.txt b/backend/python/rerankers/requirements-cpu.txt index e27a47263..5450990bd 100644 --- a/backend/python/rerankers/requirements-cpu.txt +++ b/backend/python/rerankers/requirements-cpu.txt @@ -1,4 +1,4 @@ transformers accelerate -torch==2.4.1 +torch==2.7.1+xpu 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 e27a47263..5450990bd 100644 --- a/backend/python/rerankers/requirements-cublas12.txt +++ b/backend/python/rerankers/requirements-cublas12.txt @@ -1,4 +1,4 @@ transformers accelerate -torch==2.4.1 +torch==2.7.1+xpu 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 b565a9cc1..3b5e08d01 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.9.1 +torch==2.7.1+xpu 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 7a72b3d06..ea24d8501 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+rocm6.4 +torch==2.7.1+xpu rerankers[transformers] \ No newline at end of file diff --git a/backend/python/rerankers/requirements-mps.txt b/backend/python/rerankers/requirements-mps.txt index 4064c1c5c..394116aaf 100644 --- a/backend/python/rerankers/requirements-mps.txt +++ b/backend/python/rerankers/requirements-mps.txt @@ -1,4 +1,4 @@ -torch==2.7.1 +torch==2.7.1+xpu transformers accelerate rerankers[transformers]