Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(dockerfile): fix docker build (#32)

This commit is contained in:
OlivierDehaene 2023-01-24 19:52:39 +01:00 committed by GitHub
parent 5c01e2544c
commit 13e7044ab7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -57,7 +57,7 @@ COPY server/Makefile server/Makefile
RUN cd server && make install-torch
# Install specific version of transformers
RUN cd server && make install-transformers
RUN cd server && BUILD_EXTENSIONS="True" make install-transformers
# Install server
COPY proto proto

2
server/Makefile
View File

@ -19,7 +19,7 @@ install-transformers:
install-torch:
# Install specific version of torch
pip install torch --extra-index-url https://download.pytorch.org/whl/cu116 --no-cache-dir
pip install torch --extra-index-url https://download.pytorch.org/whl/cu118 --no-cache-dir
install: gen-server install-torch install-transformers
pip install pip --upgrade