Skip to content

Commit 8a83f06

Browse files
committed
Update libtpu image global parameter in vm_resource
1 parent 976fbe8 commit 8a83f06

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

dags/common/vm_resource.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -398,5 +398,11 @@ class DockerImage(enum.Enum):
398398
"gcr.io/tpu-prod-env-multipod/maxtext_post_training_nightly:"
399399
f"{datetime.datetime.today().strftime('%Y-%m-%d')}"
400400
)
401-
TPU_OBS_LIBTPU_NIGHTLY = "us-west1-docker.pkg.dev/cienet-cmcs/tpu-obs-cloud-build/libtpu-nightly:latest"
402-
TPU_OBS_LIBTPU_STABLE = "us-west1-docker.pkg.dev/cienet-cmcs/tpu-obs-cloud-build/libtpu-stable:latest"
401+
LIBTPU_NIGHTLY = (
402+
"us-west1-docker.pkg.dev/cienet-cmcs/tpu-obs-cloud-build/libtpu-nightly:"
403+
"latest"
404+
)
405+
LIBTPU_STABLE = (
406+
"us-west1-docker.pkg.dev/cienet-cmcs/tpu-obs-cloud-build/libtpu-stable:"
407+
"latest"
408+
)

0 commit comments

Comments
 (0)