From 89236dcff8c8506c02887dc1c0838f156242c544 Mon Sep 17 00:00:00 2001
From: Eric Curtin <ecurtin@redhat.com>
Date: Mon, 11 Nov 2024 14:00:40 +0000
Subject: [PATCH] Only do dnf install for cuda images

Should fix container build for Vulkan images

Signed-off-by: Eric Curtin <ecurtin@redhat.com>
---
 container-images/scripts/build_llama_and_whisper.sh | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/container-images/scripts/build_llama_and_whisper.sh b/container-images/scripts/build_llama_and_whisper.sh
index 2a573634..bc2252bd 100644
--- a/container-images/scripts/build_llama_and_whisper.sh
+++ b/container-images/scripts/build_llama_and_whisper.sh
@@ -31,9 +31,12 @@ dnf_install() {
       "${rpm_list[@]}"
   elif [ "$containerfile" = "rocm" ]; then
     dnf install -y rocm-dev hipblas-devel rocblas-devel
-  else
+  elif [ "$containerfile" = "cuda" ]; then
     dnf install -y "${rpm_list[@]}"
   fi
+
+  # For Vulkan image, we don't need to install anything extra but rebuild with
+  # -DGGML_VULKAN
 }
 
 cmake_steps() {