Use postmerge repo for main branch

vllm-project · Dec 3, 2024 · 807266e · 807266e
1 parent 3a4b04d
commit 807266e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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" %}