diff --git a/backend/python/transformers/requirements-cpu.txt b/backend/python/transformers/requirements-cpu.txt index 5cceadec8a98..431ece0b5bb1 100644 --- a/backend/python/transformers/requirements-cpu.txt +++ b/backend/python/transformers/requirements-cpu.txt @@ -1,4 +1,4 @@ -torch==2.4.1 +torch==2.6.0+cu118 llvmlite==0.43.0 numba==0.60.0 accelerate diff --git a/backend/python/transformers/requirements-cublas11.txt b/backend/python/transformers/requirements-cublas11.txt index efaea0f6fb8f..6b2942bbb0f4 100644 --- a/backend/python/transformers/requirements-cublas11.txt +++ b/backend/python/transformers/requirements-cublas11.txt @@ -1,5 +1,5 @@ --extra-index-url https://download.pytorch.org/whl/cu118 -torch==2.4.1+cu118 +torch==2.6.0+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 55c50184ffb0..bfd99ea6c489 100644 --- a/backend/python/transformers/requirements-cublas12.txt +++ b/backend/python/transformers/requirements-cublas12.txt @@ -1,4 +1,4 @@ -torch==2.4.1 +torch==2.6.0+cu118 accelerate llvmlite==0.43.0 numba==0.60.0 diff --git a/backend/python/transformers/requirements-hipblas.txt b/backend/python/transformers/requirements-hipblas.txt index eb8713a4af75..0d6da7e9d216 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.0 -torch==2.4.1+rocm6.0 +torch==2.6.0+cu118 accelerate # https://github.com/huggingface/transformers/issues/38269#issuecomment-2900214777 transformers==4.52.1 diff --git a/backend/python/transformers/requirements-intel.txt b/backend/python/transformers/requirements-intel.txt index f9ee24a48ddd..350e414cdf71 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.3.1+cxx11.abi +torch==2.6.0+cu118 oneccl_bind_pt==2.3.100+xpu optimum[openvino] llvmlite==0.43.0