[CI] Fix docker cache invalidation by conditionally passing VLLM_MERGE_BASE_COMMIT (#230)

Signed-off-by: Amr Mahdi <[email protected]>

Author: amrmahdi

buildkite/test-template-ci.j2

@@ -276,11 +276,9 @@ steps:
         --build-arg USE_SCCACHE=1
         --build-arg TORCH_CUDA_ARCH_LIST="8.0 8.9 9.0 10.0"
         --build-arg FI_TORCH_CUDA_ARCH_LIST="8.0 8.9 9.0a 10.0a"
-        {% if branch != "main" %}
-        --build-arg VLLM_MERGE_BASE_COMMIT={{ vllm_merge_base_commit }}
         --build-arg VLLM_USE_PRECOMPILED={{ vllm_use_precompiled | default("0") }}
-        --build-arg VLLM_DOCKER_BUILD_CONTEXT=1{% if vllm_use_precompiled is defined and vllm_use_precompiled == "1" %}
-        --build-arg USE_FLASHINFER_PREBUILT_WHEEL=true{% endif %}
+        {% if vllm_use_precompiled == "1" %}
+        --build-arg VLLM_MERGE_BASE_COMMIT={{ vllm_merge_base_commit }}
         {% endif %}
         --cache-from type=registry,ref={{ cache_from }},mode=max
         --cache-to type=registry,ref={{ cache_to }},mode=max