diff --git a/docker/diffusers-doc-builder/Dockerfile b/docker/diffusers-doc-builder/Dockerfile index c9fc62707cb0..3a76b3331c17 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"]