Install compressed-tensors in Docker CPU builds

45013b60 · Daniël de Kok · bd6e8b3c · 45013b60
Commit 45013b60 authored Nov 20, 2024 by Daniël de Kok
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

Dockerfile_intel Dockerfile_intel +1 -1

No files found.
--- a/Dockerfile_intel
+++ b/Dockerfile_intel
@@ -218,7 +218,7 @@ COPY server/Makefile server/Makefile
 RUN cd server && \
    make gen-server && \
    pip install -r requirements_intel.txt && \
-    pip install ".[accelerate, peft, outlines]" --no-cache-dir
+    pip install ".[accelerate, compressed-tensors, peft, outlines]" --no-cache-dir

 # Install benchmarker
 COPY --from=builder /usr/src/target/release-opt/text-generation-benchmark /usr/local/bin/text-generation-benchmark