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

13e7044a · OlivierDehaene · GitHub · 5c01e254 · 13e7044a · 13e7044a
Unverified Commit 13e7044a authored Jan 24, 2023 by OlivierDehaene Committed by GitHub Jan 24, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

Dockerfile Dockerfile +1 -1

server/Makefile server/Makefile +1 -1

No files found.
--- a/Dockerfile
+++ b/Dockerfile
@@ -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

--- a/server/Makefile
+++ b/server/Makefile
@@ -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