Skip to content

Commit be23f7d

Browse files
authored
[Docker] update doc builder dockerfile to include quant libs. (#11728)
update doc builder dockerfile to include quant libs.
1 parent 86becea commit be23f7d

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

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)