diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index 3d708ac..24e81eb 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -5,6 +5,9 @@ FROM ubuntu:26.04
 ENV PIP_BREAK_SYSTEM_PACKAGES=1
 ENV PYTHONUNBUFFERED=1
 
+# Copy compiled ROCm from builder
+COPY --from=builder /opt/rocm /opt/rocm
+
 RUN apt update -y && apt install -y --no-install-recommends \
     rocm-dev \
     rocm-libs \
@@ -18,7 +21,6 @@ RUN apt update -y && apt install -y --no-install-recommends \
     libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Set working directory
 WORKDIR /app
 
 # Clone ComfyUI
@@ -28,17 +30,11 @@ RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
 RUN grep -vE 'torch|torchaudio|torchvision' requirements.txt > req_no_torch.txt \
     && pip3 install --no-cache-dir -r req_no_torch.txt
 
-# Set custom Nodes
+# Setup Manager
 WORKDIR /app/custom_nodes
-
-# Install ComfyUi Manager
 RUN git clone https://github.com/Comfy-Org/ComfyUI-Manager.git
 
-# Go back to root
 WORKDIR /app
-
-# Expose the default ComfyUI port
 EXPOSE 8188
 
-# Launch ComfyUI
-CMD ["python3", "main.py", "--listen", "0.0.0.0"]
+CMD ["python3", "main.py", "--listen", "0.0.0.0"]
\ No newline at end of file