From 69f414a3be694ea8d919396b9f939a03475cae01 Mon Sep 17 00:00:00 2001 From: Jeff MAURY Date: Tue, 20 Aug 2024 08:48:40 +0200 Subject: [PATCH] fix: update Vulkan flag Signed-off-by: Jeff MAURY --- chat/vulkan/amd64/Containerfile | 2 +- chat/vulkan/arm64/Containerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/chat/vulkan/amd64/Containerfile b/chat/vulkan/amd64/Containerfile index 8cc796d..31c9097 100644 --- a/chat/vulkan/amd64/Containerfile +++ b/chat/vulkan/amd64/Containerfile @@ -30,7 +30,7 @@ RUN dnf install -y python3-dnf-plugin-versionlock && \ WORKDIR /locallm COPY requirements.txt ./ RUN pip install --upgrade pip -ENV CMAKE_ARGS="-DLLAMA_VULKAN=on" +ENV CMAKE_ARGS="-DGGML_VULKAN=on" ENV FORCE_CMAKE=1 RUN pip install --target=/locallm --no-cache-dir --upgrade -r requirements.txt diff --git a/chat/vulkan/arm64/Containerfile b/chat/vulkan/arm64/Containerfile index 6456b0f..104bea2 100644 --- a/chat/vulkan/arm64/Containerfile +++ b/chat/vulkan/arm64/Containerfile @@ -29,7 +29,7 @@ https://dl.fedoraproject.org/pub/epel/epel-release-latest-9.noarch.rpm && \ dnf clean all COPY requirements.txt ./ RUN pip install --upgrade pip -ENV CMAKE_ARGS="-DLLAMA_VULKAN=on" +ENV CMAKE_ARGS="-DGGML_VULKAN=on" ENV FORCE_CMAKE=1 RUN pip install --target=/locallm --no-cache-dir --upgrade -r requirements.txt