diff --git a/Dockerfile b/Dockerfile
index 2d7e08e3..c2965b35 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -51,9 +51,9 @@ RUN uv pip install comfy-cli pip setuptools wheel
 
 # Install ComfyUI
 RUN if [ -n "${CUDA_VERSION_FOR_COMFY}" ]; then \
-      /usr/bin/yes | comfy --workspace /comfyui install --version "${COMFYUI_VERSION}" --cuda-version "${CUDA_VERSION_FOR_COMFY}" --nvidia; \
+      comfy --skip-prompt --workspace /comfyui install --version "${COMFYUI_VERSION}" --cuda-version "${CUDA_VERSION_FOR_COMFY}" --nvidia; \
     else \
-      /usr/bin/yes | comfy --workspace /comfyui install --version "${COMFYUI_VERSION}" --nvidia; \
+      comfy --skip-prompt --workspace /comfyui install --version "${COMFYUI_VERSION}" --nvidia; \
     fi
 
 # Upgrade PyTorch if needed (for newer CUDA versions)
@@ -137,4 +137,4 @@ RUN if [ "$MODEL_TYPE" = "flux1-dev-fp8" ]; then \
 FROM base AS final
 
 # Copy models from stage 2 to the final image
-COPY --from=downloader /comfyui/models /comfyui/models
\ No newline at end of file
+COPY --from=downloader /comfyui/models /comfyui/models