From 33bebd5114fd38847b296170e20a1fd774a4a096 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Tue, 5 Aug 2025 14:47:01 +0200 Subject: [PATCH] chore(deps): bump torch and diffusers (#5970) * chore(ci): add backend build tests Signed-off-by: Ettore Di Giacinto * chore(deps): bump torch and diffusers Signed-off-by: Ettore Di Giacinto * chore(ci): run diffusers/hipblas on self-hosted Signed-off-by: Ettore Di Giacinto * chore(ci): do not publish darwin if building from PRs Signed-off-by: Ettore Di Giacinto --------- Signed-off-by: Ettore Di Giacinto --- .github/workflows/backend.yml | 5 ++++- backend/python/diffusers/requirements-cpu.txt | 4 ++-- backend/python/diffusers/requirements-cublas11.txt | 4 ++-- backend/python/diffusers/requirements-cublas12.txt | 4 ++-- backend/python/diffusers/requirements-hipblas.txt | 8 ++++---- backend/python/diffusers/requirements-intel.txt | 8 ++++---- 6 files changed, 18 insertions(+), 15 deletions(-) diff --git a/.github/workflows/backend.yml b/.github/workflows/backend.yml index ee8d2a53c..479dbba17 100644 --- a/.github/workflows/backend.yml +++ b/.github/workflows/backend.yml @@ -2,6 +2,7 @@ name: 'build backend container images' on: + pull_request: push: branches: - master @@ -325,7 +326,7 @@ jobs: platforms: 'linux/amd64' tag-latest: 'auto' tag-suffix: '-gpu-rocm-hipblas-diffusers' - runs-on: 'ubuntu-latest' + runs-on: 'arc-runner-set' base-image: "rocm/dev-ubuntu-22.04:6.1" skip-drivers: 'false' backend: "diffusers" @@ -955,6 +956,7 @@ jobs: path: build/llama-cpp.tar llama-cpp-darwin-publish: needs: llama-cpp-darwin + if: github.event_name != 'pull_request' runs-on: ubuntu-latest steps: - name: Download llama-cpp.tar @@ -1043,6 +1045,7 @@ jobs: name: llama-cpp-tar-x86 path: build/llama-cpp.tar llama-cpp-darwin-x86-publish: + if: github.event_name != 'pull_request' needs: llama-cpp-darwin-x86 runs-on: ubuntu-latest steps: diff --git a/backend/python/diffusers/requirements-cpu.txt b/backend/python/diffusers/requirements-cpu.txt index 20667cc04..7b19bf219 100644 --- a/backend/python/diffusers/requirements-cpu.txt +++ b/backend/python/diffusers/requirements-cpu.txt @@ -1,9 +1,9 @@ -diffusers +git+https://github.com/huggingface/diffusers opencv-python transformers accelerate compel peft sentencepiece -torch==2.4.1 +torch==2.7.1 optimum-quanto \ No newline at end of file diff --git a/backend/python/diffusers/requirements-cublas11.txt b/backend/python/diffusers/requirements-cublas11.txt index 19e2d72e3..ce2d10c49 100644 --- a/backend/python/diffusers/requirements-cublas11.txt +++ b/backend/python/diffusers/requirements-cublas11.txt @@ -1,6 +1,6 @@ --extra-index-url https://download.pytorch.org/whl/cu118 -torch==2.4.1+cu118 -diffusers +torch==2.7.1+cu118 +git+https://github.com/huggingface/diffusers opencv-python transformers accelerate diff --git a/backend/python/diffusers/requirements-cublas12.txt b/backend/python/diffusers/requirements-cublas12.txt index 3992b0395..af5daf804 100644 --- a/backend/python/diffusers/requirements-cublas12.txt +++ b/backend/python/diffusers/requirements-cublas12.txt @@ -1,5 +1,5 @@ -torch==2.4.1 -diffusers +torch==2.7.1 +git+https://github.com/huggingface/diffusers opencv-python transformers accelerate diff --git a/backend/python/diffusers/requirements-hipblas.txt b/backend/python/diffusers/requirements-hipblas.txt index 17cf72491..2bab13494 100644 --- a/backend/python/diffusers/requirements-hipblas.txt +++ b/backend/python/diffusers/requirements-hipblas.txt @@ -1,7 +1,7 @@ ---extra-index-url https://download.pytorch.org/whl/rocm6.0 -torch==2.3.1+rocm6.0 -torchvision==0.18.1+rocm6.0 -diffusers +--extra-index-url https://download.pytorch.org/whl/rocm6.3 +torch==2.7.1+rocm6.3 +torchvision==0.22.1+rocm6.3 +git+https://github.com/huggingface/diffusers opencv-python transformers accelerate diff --git a/backend/python/diffusers/requirements-intel.txt b/backend/python/diffusers/requirements-intel.txt index eb7448b03..98ccac7b6 100644 --- a/backend/python/diffusers/requirements-intel.txt +++ b/backend/python/diffusers/requirements-intel.txt @@ -1,11 +1,11 @@ --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 -torchvision==0.18.1+cxx11.abi -oneccl_bind_pt==2.3.100+xpu +torch==2.5.1+cxx11.abi +torchvision==0.20.1+cxx11.abi +oneccl_bind_pt==2.8.0+xpu optimum[openvino] setuptools -diffusers +git+https://github.com/huggingface/diffusers opencv-python transformers accelerate