more quiet

stanford-crfm · Sep 6, 2024 · 3d8cee4 · 3d8cee4
1 parent 574eb54
commit 3d8cee4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/levanter/infra/tpus.py b/src/levanter/infra/tpus.py
@@ -79,6 +79,8 @@ def describe_tpu(tpu_name, zone):
 
 
 def start_tpu_vm(tpu_name, *, tpu_type, capacity_type, version, zone, node_count):
+    # ensure alpha is enabled
+    run_command("gcloud", "components", "install", "alpha", "--quiet")
     if version is None:
         version = "tpu-ubuntu2204-base"
     tpu_stat = describe_tpu(tpu_name, zone)