From 9087ddc4de0e5f0d3c0eb8432d3e3092d77a0901 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Tue, 5 Aug 2025 19:45:20 +0200 Subject: [PATCH] chore(deps): bump torch and sentence-transformers (#5969) * chore(deps): bump torch and sentence-transformers Signed-off-by: Ettore Di Giacinto * chore(ci): add backend build tests Signed-off-by: Ettore Di Giacinto * chore: move jobs to self-hosted Signed-off-by: Ettore Di Giacinto --------- Signed-off-by: Ettore Di Giacinto --- .github/workflows/backend.yml | 2 +- backend/python/transformers/requirements-cpu.txt | 4 ++-- backend/python/transformers/requirements-cublas11.txt | 4 ++-- backend/python/transformers/requirements-cublas12.txt | 4 ++-- backend/python/transformers/requirements-hipblas.txt | 6 +++--- backend/python/transformers/requirements-intel.txt | 6 +++--- 6 files changed, 13 insertions(+), 13 deletions(-) diff --git a/.github/workflows/backend.yml b/.github/workflows/backend.yml index 479dbba17..0142018c7 100644 --- a/.github/workflows/backend.yml +++ b/.github/workflows/backend.yml @@ -314,7 +314,7 @@ jobs: platforms: 'linux/amd64' tag-latest: 'auto' tag-suffix: '-gpu-rocm-hipblas-transformers' - runs-on: 'ubuntu-latest' + runs-on: 'arc-runner-set' base-image: "rocm/dev-ubuntu-22.04:6.1" skip-drivers: 'false' backend: "transformers" diff --git a/backend/python/transformers/requirements-cpu.txt b/backend/python/transformers/requirements-cpu.txt index 065c5c2d9..c944de271 100644 --- a/backend/python/transformers/requirements-cpu.txt +++ b/backend/python/transformers/requirements-cpu.txt @@ -1,9 +1,9 @@ -torch==2.4.1 +torch==2.7.1 llvmlite==0.43.0 numba==0.60.0 accelerate transformers bitsandbytes outetts -sentence-transformers==3.4.1 +sentence-transformers==5.0.0 protobuf==6.31.0 \ No newline at end of file diff --git a/backend/python/transformers/requirements-cublas11.txt b/backend/python/transformers/requirements-cublas11.txt index e0f6edaf4..db68ca957 100644 --- a/backend/python/transformers/requirements-cublas11.txt +++ b/backend/python/transformers/requirements-cublas11.txt @@ -1,10 +1,10 @@ --extra-index-url https://download.pytorch.org/whl/cu118 -torch==2.4.1+cu118 +torch==2.7.1+cu118 llvmlite==0.43.0 numba==0.60.0 accelerate transformers bitsandbytes outetts -sentence-transformers==4.1.0 +sentence-transformers==5.0.0 protobuf==6.31.0 \ No newline at end of file diff --git a/backend/python/transformers/requirements-cublas12.txt b/backend/python/transformers/requirements-cublas12.txt index ff5f9b40b..67783fcee 100644 --- a/backend/python/transformers/requirements-cublas12.txt +++ b/backend/python/transformers/requirements-cublas12.txt @@ -1,9 +1,9 @@ -torch==2.4.1 +torch==2.7.1 accelerate llvmlite==0.43.0 numba==0.60.0 transformers bitsandbytes outetts -sentence-transformers==4.1.0 +sentence-transformers==5.0.0 protobuf==6.31.0 \ No newline at end of file diff --git a/backend/python/transformers/requirements-hipblas.txt b/backend/python/transformers/requirements-hipblas.txt index ceae62cf9..534330bac 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 +--extra-index-url https://download.pytorch.org/whl/rocm6.3 +torch==2.7.1+rocm6.3 accelerate transformers llvmlite==0.43.0 @@ -7,5 +7,5 @@ numba==0.60.0 bitsandbytes outetts bitsandbytes -sentence-transformers==4.1.0 +sentence-transformers==5.0.0 protobuf==6.31.0 \ No newline at end of file diff --git a/backend/python/transformers/requirements-intel.txt b/backend/python/transformers/requirements-intel.txt index cdebe3267..22e2208ce 100644 --- a/backend/python/transformers/requirements-intel.txt +++ b/backend/python/transformers/requirements-intel.txt @@ -1,7 +1,7 @@ --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 -oneccl_bind_pt==2.3.100+xpu +torch==2.5.1+cxx11.abi +oneccl_bind_pt==2.8.0+xpu optimum[openvino] llvmlite==0.43.0 numba==0.60.0 @@ -9,5 +9,5 @@ transformers intel-extension-for-transformers bitsandbytes outetts -sentence-transformers==4.1.0 +sentence-transformers==5.0.0 protobuf==6.31.0 \ No newline at end of file