xiaohei/model_garden
1
0
Fork 0
mirror of https://github.com/GoogleCloudPlatform/vertex-ai-samples.git synced 2026-09-26 14:42:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support checking H100 quota (#3367)

Browse Source 
Co-authored-by: minwoopark <minwoopark@google.com>
This commit is contained in:
Minwoo Park
2024-07-31 01:59:09 +00:00
committed by GitHub
co-authored by minwoopark
parent ba526e4ac4
commit c5443d656c
1 changed files with 2 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
community-content/vertex_model_garden/model_oss/notebook_util/common_util.py
+2 -16
View File
@@ -231,22 +231,6 @@ def download_image(url: str) -> str:
return Image.open(io.BytesIO(response.content))
def resize_image(image: Any, new_width: int = 1000) -> Any:
"""Resizes an image to a certain width.
Args:
image: The image which has to be resized.
new_width: New width of the image.
Returns:
New resized image.
"""
width, height = image.size
new_height = int(height * new_width / width)
new_img = image.resize((new_width, new_height))
return new_img
def load_img(path: str) -> Any:
"""Reads image from path and return PIL.Image instance.
@@ -389,6 +373,7 @@ def get_resource_id(accelerator_type: str, is_for_training: bool) -> str:
"NVIDIA_L4": "custom_model_training_nvidia_l4_gpus",
"NVIDIA_TESLA_A100": "custom_model_training_nvidia_a100_gpus",
"NVIDIA_A100_80GB": "custom_model_training_nvidia_a100_80gb_gpus",
"NVIDIA_H100_80GB": "custom_model_training_nvidia_h100_gpus",
"NVIDIA_TESLA_T4": "custom_model_training_nvidia_t4_gpus",
"TPU_V5e": "custom_model_training_tpu_v5e",
"TPU_V3": "custom_model_training_tpu_v3",
@@ -398,6 +383,7 @@ def get_resource_id(accelerator_type: str, is_for_training: bool) -> str:
"NVIDIA_L4": "custom_model_serving_nvidia_l4_gpus",
"NVIDIA_TESLA_A100": "custom_model_serving_nvidia_a100_gpus",
"NVIDIA_A100_80GB": "custom_model_serving_nvidia_a100_80gb_gpus",
"NVIDIA_H100_80GB": "custom_model_serving_nvidia_h100_gpus",
"NVIDIA_TESLA_T4": "custom_model_serving_nvidia_t4_gpus",
"TPU_V5e": "custom_model_serving_tpu_v5e",
}