diff --git a/pathwaysutils/experimental/shared_pathways_service/deploy_pathways_service.py b/pathwaysutils/experimental/shared_pathways_service/deploy_pathways_service.py index c18b120..34b154f 100644 --- a/pathwaysutils/experimental/shared_pathways_service/deploy_pathways_service.py +++ b/pathwaysutils/experimental/shared_pathways_service/deploy_pathways_service.py @@ -113,7 +113,7 @@ def get_tpu_config(tpu_type: str) -> TPUConfig: "tpu7x": TPUConfig( machine_type="tpu7x-standard-4t", chips_per_vm=4, - accelerator_label="tpu-v7-slice", + accelerator_label="tpu7x", instance_prefix="tpu7x", ), } diff --git a/pathwaysutils/experimental/shared_pathways_service/validators.py b/pathwaysutils/experimental/shared_pathways_service/validators.py index 818aaba..7b28440 100644 --- a/pathwaysutils/experimental/shared_pathways_service/validators.py +++ b/pathwaysutils/experimental/shared_pathways_service/validators.py @@ -68,7 +68,7 @@ def _validate_tpu_supported(tpu_instance_with_topology: str) -> None: # tpuv6e:2x4 -> type='tpuv6e', topology='2x4' # tpuv5:2x2x1 -> type='tpuv5', topology='2x2x1' match = re.match( - r"^(?:tpuv(?:5e|5|6e)):(?P\d+(?:x\d+){1,2})$", + r"^(?:tpu(?:v5e|v5|v6e|7x)):(?P\d+(?:x\d+){1,2})$", tpu_instance_with_topology, )