fix(rerankers): keep torch pins platform-specific

The CUDA 13 wheel cannot be installed by the CPU, CUDA 12, ROCm, or macOS backend builds. Keep the new cu130 pin only in the matching cublas13 requirements and restore the platform-compatible pins elsewhere. Assisted-by: Codex:gpt-5
2026-07-30 01:48:06 -04:00 · 2026-07-30 05:05:08 +00:00
parent 369e97a3c5
commit b429e28a59
4 changed files with 7 additions and 7 deletions
--- a/backend/python/rerankers/requirements-cpu.txt
+++ b/backend/python/rerankers/requirements-cpu.txt
@@ -1,4 +1,4 @@
 transformers
 accelerate
-torch==2.12.1+cu130
-rerankers[transformers]
+torch==2.7.1
+rerankers[transformers]
--- a/backend/python/rerankers/requirements-cublas12.txt
+++ b/backend/python/rerankers/requirements-cublas12.txt
@@ -1,4 +1,4 @@
 transformers
 accelerate
-torch==2.12.1+cu130
-rerankers[transformers]
+torch==2.7.1
+rerankers[transformers]
--- 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/rocm7.0
 transformers
 accelerate
-torch==2.12.1+cu130
-rerankers[transformers]
+torch==2.10.0+rocm7.0
+rerankers[transformers]
--- a/backend/python/rerankers/requirements-mps.txt
+++ b/backend/python/rerankers/requirements-mps.txt
@@ -1,4 +1,4 @@
-torch==2.12.1+cu130
+torch==2.7.1
 transformers
 accelerate
 rerankers[transformers]