huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-09 19:34:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of github.com:huggingface/text-generation-inference

Browse Source
This commit is contained in:
Yannic Kilcher 2023-01-26 14:51:07 +01:00
commit 7beb968696
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