diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index 5b64a8b..a9554fe 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -30,7 +30,7 @@ WORKDIR /app
 RUN git clone --recursive https://github.com/ollama/ollama.git .
 
 # Configure and safely build with constrained parallelism to protect 96GB RAM
-RUN cmake -B build . -DOLLAMA_LLAMA_BACKENDS="vulkan" -DCMAKE_BUILD_TYPE=Release
+RUN cmake -B build . -DOLLAMA_LLAMA_BACKENDS="vulkan" -DGGML_VULKAN=ON -DCMAKE_BUILD_TYPE=Release
 RUN cmake --build build --parallel 16
 
 # Build the main Go application binary
@@ -39,6 +39,7 @@ RUN go build -o ollama .
 # Start the server
 EXPOSE 11434
 ENV OLLAMA_HOST=0.0.0.0:11434
+ENV GGML_VULKAN_DISABLE_FMA=1
 ENV OLLAMA_MODELS=/root/.ollama/models
 
 ENTRYPOINT ["/app/ollama", "serve"]
\ No newline at end of file