From 7bd402b08e1f3f21ab25ebd94d748683099cfd43 Mon Sep 17 00:00:00 2001
From: Derek Holloway <derek@mistox.com>
Date: Wed, 9 Sep 2026 19:26:06 -0700
Subject: [PATCH] use the builder for all the dependancies

---
 comfyui-rocm-dockerfile | 17 +----------------
 1 file changed, 1 insertion(+), 16 deletions(-)

diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index c5f06a1..0883903 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -36,22 +36,7 @@ RUN cmake --build build --parallel 16
 # Build the main Go application binary
 RUN go build -o ollama .
 
-# Stage 2: Minimal Runtime Image
-FROM ubuntu:24.04
-
-ENV DEBIAN_FRONTEND=noninteractive
-
-RUN apt-get update && apt-get install -y \
-    libvulkan1 \
-    mesa-vulkan-drivers \
-    vulkan-tools \
-    ca-certificates \
-    && rm -rf /var/lib/apt/lists/*
-
-# Copy the compiled Go binary and generated native backend libraries
-COPY --from=builder /app/ollama /usr/local/bin/ollama
-COPY --from=builder /app/build/lib/ollama /usr/local/lib/ollama
-
+# Start the server
 EXPOSE 11434
 ENV OLLAMA_HOST=0.0.0.0:11434
 ENV OLLAMA_MODELS=/root/.ollama/models