update dockerfile
This commit is contained in:
parent
bc4b3f97ec
commit
2ac9db513a
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue