File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -19,6 +19,7 @@ RUN apt-get update && \
19
19
libopenblas-dev \
20
20
ninja-build \
21
21
build-essential \
22
+ pkg-config \
22
23
curl
23
24
24
25
# Include global args in this stage of the build
@@ -34,13 +35,13 @@ COPY ./ ${FUNCTION_DIR}
34
35
35
36
RUN python${RUNTIME_VERSION} -m pip install --upgrade pip pytest cmake scikit-build setuptools fastapi uvicorn sse-starlette
36
37
# REF: https://github.com/abetlen/llama-cpp-python/blob/main/Dockerfile
37
- RUN cd ${FUNCTION_DIR} && CMAKE_ARGS="-DLLAMA_OPENBLAS=on " FORCE_CMAKE=1 python${RUNTIME_VERSION} setup.py develop
38
+ RUN cd ${FUNCTION_DIR} && CMAKE_ARGS="-DLLAMA_BLAS=ON -DLLAMA_BLAS_VENDOR=OpenBLAS " FORCE_CMAKE=1 python${RUNTIME_VERSION} setup.py develop
38
39
# Install Lambda Runtime Interface Client for Python
39
40
RUN python${RUNTIME_VERSION} -m pip install awslambdaric --target ${FUNCTION_DIR}
40
41
41
42
# download the model file
42
43
RUN mkdir ${FUNCTION_DIR}/model
43
- RUN curl -L https://huggingface.co/TheBloke/orca_mini_v2_7B -GGML/resolve/main/orca-mini-v2_7b .ggmlv3.q4_0.bin -o ${FUNCTION_DIR}/model/ggml-q4_0.bin
44
+ RUN curl -L https://huggingface.co/TheBloke/orca_mini_v3_7B -GGML/resolve/main/orca_mini_v3_7b .ggmlv3.q4_0.bin -o ${FUNCTION_DIR}/model/ggml-q4_0.bin
44
45
45
46
# Stage 3 - final runtime image
46
47
# Grab a fresh copy of the Python image
You can’t perform that action at this time.
0 commit comments