From e563a66114007cb5ff0938fe14b0b3b8aae30649 Mon Sep 17 00:00:00 2001 From: Rayan Dasoriya Date: Tue, 17 Jun 2025 18:31:19 +0530 Subject: [PATCH] Add support for tpu v6e quota check (#4111) Co-authored-by: Rayan Dasoriya --- .../vertex_model_garden/model_oss/notebook_util/common_util.py | 1 + 1 file changed, 1 insertion(+) diff --git a/community-content/vertex_model_garden/model_oss/notebook_util/common_util.py b/community-content/vertex_model_garden/model_oss/notebook_util/common_util.py index fb5b5c20b..f31f8e566 100644 --- a/community-content/vertex_model_garden/model_oss/notebook_util/common_util.py +++ b/community-content/vertex_model_garden/model_oss/notebook_util/common_util.py @@ -564,6 +564,7 @@ def get_resource_id( "NVIDIA_H100_MEGA_80GB": "nvidia_h100_mega_gpus", "NVIDIA_H200_141GB": "nvidia_h200_gpus", "NVIDIA_TESLA_T4": "nvidia_t4_gpus", + "TPU_V6e": "tpu_v6e", "TPU_V5e": "tpu_v5e", "TPU_V3": "tpu_v3", }