From 86becea77f3630b0e65525e4ebd2b10c26113763 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Wed, 9 Jul 2025 12:17:45 +0530 Subject: [PATCH 1/2] Pin k-diffusion for CI (#11894) * remove k-diffusion as we don't use it from the core. * Revert "remove k-diffusion as we don't use it from the core." This reverts commit 8bc86925a0b8535e85430be570f37398d211e249. * pin k-diffusion --- setup.py | 2 +- src/diffusers/dependency_versions_table.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 11b64cff2d..103ff16e35 100644 --- a/setup.py +++ b/setup.py @@ -110,7 +110,7 @@ _deps = [ "jax>=0.4.1", "jaxlib>=0.4.1", "Jinja2", - "k-diffusion>=0.0.12", + "k-diffusion==0.0.12", "torchsde", "note_seq", "librosa", diff --git a/src/diffusers/dependency_versions_table.py b/src/diffusers/dependency_versions_table.py index f35353c49e..ec52bcd636 100644 --- a/src/diffusers/dependency_versions_table.py +++ b/src/diffusers/dependency_versions_table.py @@ -17,7 +17,7 @@ deps = { "jax": "jax>=0.4.1", "jaxlib": "jaxlib>=0.4.1", "Jinja2": "Jinja2", - "k-diffusion": "k-diffusion>=0.0.12", + "k-diffusion": "k-diffusion==0.0.12", "torchsde": "torchsde", "note_seq": "note_seq", "librosa": "librosa", From be23f7df00b308456c9f8f6fdf109dad8af787a2 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Wed, 9 Jul 2025 12:27:22 +0530 Subject: [PATCH 2/2] [Docker] update doc builder dockerfile to include quant libs. (#11728) update doc builder dockerfile to include quant libs. --- docker/diffusers-doc-builder/Dockerfile | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/docker/diffusers-doc-builder/Dockerfile b/docker/diffusers-doc-builder/Dockerfile index c9fc62707c..3a76b3331c 100644 --- a/docker/diffusers-doc-builder/Dockerfile +++ b/docker/diffusers-doc-builder/Dockerfile @@ -47,6 +47,10 @@ RUN python3.10 -m pip install --no-cache-dir --upgrade pip uv==0.1.11 && \ tensorboard \ transformers \ matplotlib \ - setuptools==69.5.1 + setuptools==69.5.1 \ + bitsandbytes \ + torchao \ + gguf \ + optimum-quanto CMD ["/bin/bash"]