From be19675980d512f3b7420a549df7e179f0fa9727 Mon Sep 17 00:00:00 2001
From: Eugene Rakhmatulin <eugr@eugr.com>
Date: Fri, 30 Jan 2026 11:24:51 -0800
Subject: [PATCH] Fixed initial vllm source fetch if not using main branch

---
 Dockerfile | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 67d9748..a8cb9f3 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -141,6 +141,10 @@ RUN --mount=type=cache,id=repo-cache,target=/repo-cache \
     if [ ! -d "vllm" ]; then \
         echo "Cache miss: Cloning vLLM from scratch..." && \
         git clone --recursive https://github.com/vllm-project/vllm.git; \
+        if [ "$VLLM_REF" != "main" ]; then \
+            cd vllm && \
+            git checkout ${VLLM_REF}; \
+        fi; \
     else \
         echo "Cache hit: Fetching updates..." && \
         cd vllm && \