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