File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -20,13 +20,13 @@ RUN python3 -m pip install --upgrade pip pytest cmake scikit-build setuptools fa
20
20
21
21
# Perform the conditional installations based on the image
22
22
RUN echo "Image: ${IMAGE}" && \
23
- if [ "${IMAGE}" = "python:3-slim-bullseye " ] ; then \
23
+ if [ "${IMAGE}" = "python:3-slim-bookworm " ] ; then \
24
24
echo "OpenBLAS install:" && \
25
25
apt-get install -y --no-install-recommends libopenblas-dev && \
26
- LLAMA_OPENBLAS=1 pip install llama-cpp-python --verbose; \
26
+ CMAKE_ARGS= "-DGGML_BLAS=ON -DGGML_BLAS_VENDOR=OpenBLAS" pip install llama-cpp-python --verbose; \
27
27
else \
28
28
echo "CuBLAS install:" && \
29
- LLAMA_CUBLAS=1 pip install llama-cpp-python --verbose; \
29
+ CMAKE_ARGS= "-DGGML_CUDA=on" pip install llama-cpp-python --verbose; \
30
30
fi
31
31
32
32
# Clean up apt cache
You can’t perform that action at this time.
0 commit comments