From 45013b60a4afe53143dd316b49a85ccd96f2205b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= Date: Wed, 20 Nov 2024 14:17:47 +0000 Subject: [PATCH] Install compressed-tensors in Docker CPU builds --- Dockerfile_intel | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile_intel b/Dockerfile_intel index ea38b081ae9..fea041764c3 100644 --- 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