diff --git a/notebooks/community/model_garden/docker_source_codes/model_oss/notebook_util/common_util.py b/notebooks/community/model_garden/docker_source_codes/model_oss/notebook_util/common_util.py
index bb98d01d4..158e55879 100644
--- a/notebooks/community/model_garden/docker_source_codes/model_oss/notebook_util/common_util.py
+++ b/notebooks/community/model_garden/docker_source_codes/model_oss/notebook_util/common_util.py
@@ -543,7 +543,8 @@ def get_quota_id(
       "NVIDIA_H100_80GB": "H100GPUs",
       "NVIDIA_H100_MEGA_80GB": "H100MEGAGPUs",
       "NVIDIA_H200_141GB": "H200GPUs",
-      "NVIDIA_GB200": "B200GPUs",
+      "NVIDIA_GB200": "GB200GPUs",
+      "NVIDIA_B200": "B200GPUs",
       "NVIDIA_TESLA_T4": "T4GPUs",
       "NVIDIA_RTX_PRO_6000": "RTXPRO6000GPUs",
       "TPU_7x": "7XTPU",
diff --git a/notebooks/community/model_garden/docker_source_codes/notebook_util/common_util.py b/notebooks/community/model_garden/docker_source_codes/notebook_util/common_util.py
index bb98d01d4..158e55879 100644
--- a/notebooks/community/model_garden/docker_source_codes/notebook_util/common_util.py
+++ b/notebooks/community/model_garden/docker_source_codes/notebook_util/common_util.py
@@ -543,7 +543,8 @@ def get_quota_id(
       "NVIDIA_H100_80GB": "H100GPUs",
       "NVIDIA_H100_MEGA_80GB": "H100MEGAGPUs",
       "NVIDIA_H200_141GB": "H200GPUs",
-      "NVIDIA_GB200": "B200GPUs",
+      "NVIDIA_GB200": "GB200GPUs",
+      "NVIDIA_B200": "B200GPUs",
       "NVIDIA_TESLA_T4": "T4GPUs",
       "NVIDIA_RTX_PRO_6000": "RTXPRO6000GPUs",
       "TPU_7x": "7XTPU",