update dockerfile

This commit is contained in:
michaelfeil 2023-08-04 14:33:49 +02:00
parent bc4b3f97ec
commit 2ac9db513a
1 changed files with 1 additions and 1 deletions

View File

@ -188,7 +188,7 @@ COPY server/Makefile server/Makefile
RUN cd server && \
make gen-server && \
pip install -r requirements.txt && \
pip install ".[bnb, accelerate, ct2]" --no-cache-dir
pip install ".[bnb, accelerate, quantize, ct2]" --no-cache-dir
# Install benchmarker
COPY --from=builder /usr/src/target/release/text-generation-benchmark /usr/local/bin/text-generation-benchmark