Upload 6 files
Browse files- Dockerfile +5 -5
Dockerfile
CHANGED
|
@@ -25,14 +25,14 @@ COPY . .
|
|
| 25 |
# Setup environment variables for compilation
|
| 26 |
# We need to find where pip installed llama-cpp-python to link against it
|
| 27 |
# In docker, it's usually /usr/local/lib/python3.12/site-packages
|
| 28 |
-
|
| 29 |
|
| 30 |
# Compile the engine
|
| 31 |
# WORKDIR /app/engine
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
|
| 37 |
# Setup Model (Download during build or mount volume?
|
| 38 |
# Best practice: Download in build if small, or use script at runtime.
|
|
|
|
| 25 |
# Setup environment variables for compilation
|
| 26 |
# We need to find where pip installed llama-cpp-python to link against it
|
| 27 |
# In docker, it's usually /usr/local/lib/python3.12/site-packages
|
| 28 |
+
ENV SITE_PACKAGES=/usr/local/lib/python3.12/site-packages
|
| 29 |
|
| 30 |
# Compile the engine
|
| 31 |
# WORKDIR /app/engine
|
| 32 |
+
RUN g++ -O2 -shared -fPIC -o libbatch.so batch_server.cpp \
|
| 33 |
+
-I"${SITE_PACKAGES}/include" \
|
| 34 |
+
-L"${SITE_PACKAGES}/llama_cpp/lib" \
|
| 35 |
+
-lllama -Wl,-rpath,"${SITE_PACKAGES}/llama_cpp/lib"
|
| 36 |
|
| 37 |
# Setup Model (Download during build or mount volume?
|
| 38 |
# Best practice: Download in build if small, or use script at runtime.
|