Install compressed-tensors in Docker CPU builds

huggingface · Nov 20, 2024 · 45013b6 · 45013b6
1 parent bd6e8b3
commit 45013b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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