We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c65b0be commit 2d2ce84Copy full SHA for 2d2ce84
dags/common/vm_resource.py
@@ -62,6 +62,7 @@ class Project(enum.Enum):
62
CLOUD_TPU_INFERENCE_TEST = "cloud-tpu-inference-test"
63
TPU_PROD_ENV_LARGE_ADHOC = "tpu-prod-env-large-adhoc"
64
TPU_PROD_ENV_ONE_VM = "tpu-prod-env-one-vm"
65
+ TPU_PROD_ENV_LARGE_CONT = "tpu-prod-env-large-cont"
66
67
68
class ImageProject(enum.Enum):
@@ -233,7 +234,7 @@ class XpkClusters:
233
234
name="v5p-8-bodaborg-us-east5-a",
235
device_version=TpuVersion.V5P,
236
core_count=8,
- project=Project.CLOUD_TPU_MULTIPOD_DEV.value,
237
+ project=Project.TPU_PROD_ENV_LARGE_CONT.value,
238
zone=Zone.US_EAST5_A.value,
239
)
240
TPU_V5E_256_CLUSTER = XpkClusterConfig(
0 commit comments