diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index da9e728..a33a1fe 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -83,31 +83,11 @@ ENV CGO_ENABLED=1
 RUN cmake -B build . \
     -DOLLAMA_LLAMA_BACKENDS="vulkan" \
     -DGGML_VULKAN=ON \
+    -DGGML_CPU_ALL_VARIANTS=OFF \
     -DCMAKE_BUILD_TYPE=Release
 RUN cmake --build build --parallel 16
 RUN go build -o ollama .
 
-# Stage 3: Minimal Runtime Image
-FROM ubuntu:24.04
-
-ENV DEBIAN_FRONTEND=noninteractive
-
-RUN apt-get update && apt-get install -y \
-    ca-certificates \
-    libvulkan1 \
-    vulkan-tools \
-    libexpat1 \
-    libelf1 \
-    libdrm2 \
-    && rm -rf /var/lib/apt/lists/*
-
-# Copy compiled Mesa-git libraries and ICD files from Stage 1
-COPY --from=builder /opt/mesa /opt/mesa
-
-# Copy compiled Ollama binary from Stage 2
-COPY --from=builder /app/ollama /app/ollama
-COPY --from=builder /app/build /app/build
-
 EXPOSE 11434
 
 ENV OLLAMA_HOST=0.0.0.0:11434