x
This commit is contained in:
+5
-2
@@ -67,8 +67,11 @@ ENV FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE"
|
|||||||
RUN pip install flash-attn --no-build-isolation \
|
RUN pip install flash-attn --no-build-isolation \
|
||||||
--index-url https://pypi.tuna.tsinghua.edu.cn/simple
|
--index-url https://pypi.tuna.tsinghua.edu.cn/simple
|
||||||
|
|
||||||
# 6. Clone vLLM
|
# 6. Clone vLLM (with retry mechanism)
|
||||||
RUN git clone https://github.com/vllm-project/vllm.git /opt/vllm
|
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
|
WORKDIR /opt/vllm
|
||||||
|
|
||||||
# --- PATCHING ---
|
# --- PATCHING ---
|
||||||
|
|||||||
Reference in New Issue
Block a user