Karan6933 commited on
Commit
70f4ee4
·
verified ·
1 Parent(s): 65a1dcc

Upload 6 files

Browse files
Files changed (1) hide show
  1. 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
- # 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.
 
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.