Skip to content

Commit 72f6bac

Browse files
committed
update doc builder dockerfile to include quant libs.
1 parent 9b834f8 commit 72f6bac

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

docker/diffusers-doc-builder/Dockerfile

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,10 @@ RUN python3.10 -m pip install --no-cache-dir --upgrade pip uv==0.1.11 && \
4747
tensorboard \
4848
transformers \
4949
matplotlib \
50-
setuptools==69.5.1
50+
setuptools==69.5.1 \
51+
bitsandbytes \
52+
torchao \
53+
gguf \
54+
optimum-quanto
5155

5256
CMD ["/bin/bash"]

0 commit comments

Comments
 (0)