diff --git a/backend/python/rerankers/requirements-cpu.txt b/backend/python/rerankers/requirements-cpu.txt index 1225fa4db5a5..d26c5a599baf 100644 --- a/backend/python/rerankers/requirements-cpu.txt +++ b/backend/python/rerankers/requirements-cpu.txt @@ -1,4 +1,4 @@ transformers accelerate torch==2.7.1 -rerankers[transformers] \ No newline at end of file +rerankers[transformers] diff --git a/backend/python/rerankers/requirements-cublas12.txt b/backend/python/rerankers/requirements-cublas12.txt index 1225fa4db5a5..d26c5a599baf 100644 --- a/backend/python/rerankers/requirements-cublas12.txt +++ b/backend/python/rerankers/requirements-cublas12.txt @@ -1,4 +1,4 @@ transformers accelerate torch==2.7.1 -rerankers[transformers] \ No newline at end of file +rerankers[transformers] diff --git a/backend/python/rerankers/requirements-cublas13.txt b/backend/python/rerankers/requirements-cublas13.txt index b565a9cc154a..7ccb11fe4e5b 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.12.1+cu130 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 3f4a21d9a20d..f95d03c6f6b3 100644 --- a/backend/python/rerankers/requirements-hipblas.txt +++ b/backend/python/rerankers/requirements-hipblas.txt @@ -2,4 +2,4 @@ transformers accelerate torch==2.10.0+rocm7.0 -rerankers[transformers] \ No newline at end of file +rerankers[transformers]