diff --git a/backend/python/transformers/requirements-cpu.txt b/backend/python/transformers/requirements-cpu.txt index 655ee645f5fd..1548234678c5 100644 --- a/backend/python/transformers/requirements-cpu.txt +++ b/backend/python/transformers/requirements-cpu.txt @@ -1,4 +1,4 @@ -torch==2.7.1 +torch==2.7.1+cu118 llvmlite==0.43.0 numba==0.60.0 accelerate diff --git a/backend/python/transformers/requirements-cublas12.txt b/backend/python/transformers/requirements-cublas12.txt index ae8e8197e360..08d8f1144697 100644 --- a/backend/python/transformers/requirements-cublas12.txt +++ b/backend/python/transformers/requirements-cublas12.txt @@ -1,4 +1,4 @@ -torch==2.7.1 +torch==2.7.1+cu118 accelerate llvmlite==0.43.0 numba==0.60.0 diff --git a/backend/python/transformers/requirements-cublas13.txt b/backend/python/transformers/requirements-cublas13.txt index 2e2dda6b8b84..9ea441dc2b69 100644 --- a/backend/python/transformers/requirements-cublas13.txt +++ b/backend/python/transformers/requirements-cublas13.txt @@ -1,5 +1,5 @@ --extra-index-url https://download.pytorch.org/whl/cu130 -torch==2.9.0 +torch==2.7.1+cu118 llvmlite==0.43.0 numba==0.60.0 transformers diff --git a/backend/python/transformers/requirements-hipblas.txt b/backend/python/transformers/requirements-hipblas.txt index 2b468314cbf2..c84c691230c5 100644 --- a/backend/python/transformers/requirements-hipblas.txt +++ b/backend/python/transformers/requirements-hipblas.txt @@ -1,5 +1,5 @@ --extra-index-url https://download.pytorch.org/whl/rocm6.3 -torch==2.7.1+rocm6.3 +torch==2.7.1+cu118 accelerate transformers llvmlite==0.43.0 diff --git a/backend/python/transformers/requirements-intel.txt b/backend/python/transformers/requirements-intel.txt index e0d31d2fd38b..3f552dcb8df3 100644 --- a/backend/python/transformers/requirements-intel.txt +++ b/backend/python/transformers/requirements-intel.txt @@ -1,6 +1,6 @@ --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ intel-extension-for-pytorch==2.3.110+xpu -torch==2.5.1+cxx11.abi +torch==2.7.1+cu118 oneccl_bind_pt==2.8.0+xpu optimum[openvino] llvmlite==0.43.0