Add hostname GCP

Signed-off-by: kevin <[email protected]>
vllm-project · Sep 24, 2024 · 2ca807e · 2ca807e
1 parent 24f4deb
commit 2ca807e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/infra/gcp_old/main.tf b/infra/gcp_old/main.tf
@@ -75,6 +75,7 @@ resource "google_tpu_v2_vm" "tpu" {
       sudo -u buildkite-agent gcloud auth configure-docker us-central1-docker.pkg.dev --quiet
 
       sudo sed -i "s/xxx/${var.buildkite_agent_token}/g" /etc/buildkite-agent/buildkite-agent.cfg
+      sudo sed -i 's/name="%hostname-%spawn"/name="vllm-tpu-${count.index}"/' /etc/buildkite-agent/buildkite-agent.cfg
       echo 'tags="queue=tpu"' | sudo tee -a /etc/buildkite-agent/buildkite-agent.cfg
       echo 'HF_TOKEN=${var.huggingface_token}' | sudo tee -a /etc/environment