From 6f508237aafbcb07fd89076d57d46bf51f15cd8f Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Mon, 19 Oct 2026 17:10:25 +0530 Subject: [PATCH] Fix PyArrow compatibility in CPU CI --- .github/workflows/pr_tests.yml | 2 +- .github/workflows/push_tests_fast.yml | 2 +- docker/diffusers-pytorch-cpu/Dockerfile | 1 + 3 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.github/workflows/pr_tests.yml b/.github/workflows/pr_tests.yml index b25907d34320..9fe6ac29cd08 100644 --- a/.github/workflows/pr_tests.yml +++ b/.github/workflows/pr_tests.yml @@ -133,7 +133,7 @@ jobs: - name: Install dependencies run: | printf 'torch==2.10.0\ntorchvision==0.25.0\ntorchaudio==2.10.0\n' > "$UV_OVERRIDE" - uv pip install -e ".[quality]" + uv pip install -e ".[quality]" "pyarrow<26" uv pip uninstall transformers huggingface_hub && UV_PRERELEASE=allow uv pip install -U transformers@git+https://github.com/huggingface/transformers.git uv pip uninstall accelerate && uv pip install -U accelerate@git+https://github.com/huggingface/accelerate.git --no-deps diff --git a/.github/workflows/push_tests_fast.yml b/.github/workflows/push_tests_fast.yml index 5a0dbcd2c8b7..42666034a8f8 100644 --- a/.github/workflows/push_tests_fast.yml +++ b/.github/workflows/push_tests_fast.yml @@ -68,7 +68,7 @@ jobs: - name: Install dependencies run: | - uv pip install -e ".[quality]" + uv pip install -e ".[quality]" "pyarrow<26" - name: Environment run: | diff --git a/docker/diffusers-pytorch-cpu/Dockerfile b/docker/diffusers-pytorch-cpu/Dockerfile index e53b4ccd536b..7703883a567e 100644 --- a/docker/diffusers-pytorch-cpu/Dockerfile +++ b/docker/diffusers-pytorch-cpu/Dockerfile @@ -31,6 +31,7 @@ RUN uv pip install --no-cache-dir "git+https://github.com/huggingface/diffusers. RUN uv pip install --no-cache-dir \ accelerate \ numpy==1.26.4 \ + "pyarrow<26" \ hf_xet RUN apt-get clean && rm -rf /var/lib/apt/lists/* && apt-get autoremove && apt-get autoclean