fix: update Vulkan flag

Signed-off-by: Jeff MAURY <[email protected]>
containers · Aug 20, 2024 · 69f414a · 69f414a
1 parent 3523c65
commit 69f414a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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