From 807266e6a13991a4b397f9562c46d1e93ac01139 Mon Sep 17 00:00:00 2001
From: "Kevin H. Luu" <kevin@anyscale.com>
Date: Tue, 3 Dec 2024 13:20:56 -0800
Subject: [PATCH] Use postmerge repo for main branch

---
 scripts/test-template-aws.j2 | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/scripts/test-template-aws.j2 b/scripts/test-template-aws.j2
index 6068fe9..ef98bc7 100644
--- a/scripts/test-template-aws.j2
+++ b/scripts/test-template-aws.j2
@@ -1,4 +1,7 @@
 {% set docker_image = "public.ecr.aws/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT" %}
+{% if branch == "main" %}
+{% set docker_image = "public.ecr.aws/q9t5s3a7/vllm-ci-postmerge-repo:$BUILDKITE_COMMIT" %}
+{% endif %}
 {% set docker_image_amd = "rocm/vllm-ci:$BUILDKITE_COMMIT" %}
 {% set default_working_dir = "/vllm-workspace/tests" %}
 {% set hf_home = "/root/.cache/huggingface" %}