File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 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
2020
2121# Perform the conditional installations based on the image
2222RUN echo "Image: ${IMAGE}" && \
23- if [ "${IMAGE}" = "python:3-slim-bullseye " ] ; then \
23+ if [ "${IMAGE}" = "python:3-slim-bookworm " ] ; then \
2424 echo "OpenBLAS install:" && \
2525 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; \
2727else \
2828 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; \
3030fi
3131
3232# Clean up apt cache
You can’t perform that action at this time.
0 commit comments