Only do dnf install for cuda images

Should fix container build for Vulkan images Signed-off-by: Eric Curtin <[email protected]>
containers · Nov 11, 2024 · 89236dc · 89236dc
1 parent 31ae27b
commit 89236dc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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() {