alihan/text-generation-inference
1
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2023-08-23 10:47:54 +03:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
OlivierDehaene
2023-01-24 19:52:39 +01:00
committed by GitHub
parent 5c01e2544c
commit 13e7044ab7
2 changed files with 2 additions and 2 deletions
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