diff --git a/Dockerfile b/Dockerfile index 09ab916..5bcf4df 100644 --- a/Dockerfile +++ b/Dockerfile @@ -67,8 +67,11 @@ ENV FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE" RUN pip install flash-attn --no-build-isolation \ --index-url https://pypi.tuna.tsinghua.edu.cn/simple -# 6. Clone vLLM -RUN git clone https://github.com/vllm-project/vllm.git /opt/vllm +# 6. Clone vLLM (with retry mechanism) +RUN for i in 1 2 3; do \ + git clone --depth 1 https://github.com/vllm-project/vllm.git /opt/vllm && break || \ + { echo "Attempt $i failed. Retrying in 5 seconds..."; sleep 5; rm -rf /opt/vllm 2>/dev/null; }; \ + done WORKDIR /opt/vllm # --- PATCHING ---