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

Set system_labels in notebooks

Browse Source 
PiperOrigin-RevId: 713040032
This commit is contained in:
Dustin Luong
2025-01-07 14:20:25 -08:00
committed by Copybara-Service
parent 847a49f6d0
commit 883e1e5fc1
81 changed files with 211 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
notebooks/community/model_garden/model_garden_e5.ipynb
+3
View File
@@ -220,6 +220,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_e5.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_jax_dito.ipynb
+3
View File
@@ -1336,6 +1336,9 @@
" machine_type=\"n1-highmem-16\",\n",
" min_replica_count=1,\n",
" max_replica_count=1,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_jax_dito.ipynb\"\n",
" },\n",
")"
]
},
notebooks/community/model_garden/model_garden_jax_fvlm.ipynb
+3
View File
@@ -1338,6 +1338,9 @@
" machine_type=\"n1-highmem-16\",\n",
" min_replica_count=1,\n",
" max_replica_count=1,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_jax_fvlm.ipynb\"\n",
" },\n",
")"
]
},
notebooks/community/model_garden/model_garden_jax_owl_vit_v2.ipynb
+1
View File
@@ -335,6 +335,7 @@
" machine_type=\"n1-highmem-64\",\n",
" min_replica_count=1,\n",
" max_replica_count=1,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_jax_owl_vit_v2.ipynb\"},\n",
")\n",
"\n",
"# @markdown Click \"Show Code\" to see more details."
notebooks/community/model_garden/model_garden_jax_paligemma_deployment.ipynb
+1
View File
@@ -391,6 +391,7 @@
" enable_access_logging=True,\n",
" min_replica_count=1,\n",
" sync=True,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_jax_paligemma_deployment.ipynb\"},\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_jax_paligemma_finetuning.ipynb
+1
View File
@@ -570,6 +570,7 @@
" enable_access_logging=True,\n",
" min_replica_count=1,\n",
" sync=True,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_jax_paligemma_finetuning.ipynb\"},\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_jax_stable_diffusion_xl.ipynb
+3
View File
@@ -286,6 +286,9 @@
" enable_access_logging=True,\n",
" min_replica_count=1,\n",
" sync=True,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_jax_stable_diffusion_xl.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_jax_vision_transformer.ipynb
+3
View File
@@ -795,6 +795,9 @@
" accelerator_count=1,\n",
" min_replica_count=1,\n",
" max_replica_count=1,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_jax_vision_transformer.ipynb\"\n",
" },\n",
")"
]
},
notebooks/community/model_garden/model_garden_keras_stable_diffusion.ipynb
+3
View File
@@ -370,6 +370,9 @@
" max_replica_count=1,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_keras_stable_diffusion.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_keras_yolov8.ipynb
+1
View File
@@ -513,6 +513,7 @@
" accelerator_count=1,\n",
" min_replica_count=1,\n",
" max_replica_count=1,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_keras_yolov8.ipynb\"},\n",
")\n",
"\n",
"\n",
notebooks/community/model_garden/model_garden_mammut.ipynb
+3
View File
@@ -245,6 +245,9 @@
" accelerator_count=accelerator_count,\n",
" min_replica_count=1,\n",
" max_replica_count=1,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_mammut.ipynb\"\n",
" },\n",
" )\n",
"\n",
" print(\"The deployed job name is: \", deploy_model_name)\n",
notebooks/community/model_garden/model_garden_mediapipe_image_generation.ipynb
+3
View File
@@ -898,6 +898,9 @@
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_mediapipe_image_generation.ipynb\"\n",
" },\n",
")"
]
},
notebooks/community/model_garden/model_garden_proprietary_image_classification.ipynb
+3
View File
@@ -541,6 +541,9 @@
" traffic_split={\"0\": 100},\n",
" min_replica_count=1,\n",
" max_replica_count=1,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_proprietary_image_classification.ipynb\"\n",
" },\n",
")\n",
"\n",
"endpoint_id = endpoint.name\n",
notebooks/community/model_garden/model_garden_proprietary_image_object_detection.ipynb
+3
View File
@@ -614,6 +614,9 @@
" traffic_split={\"0\": 100},\n",
" min_replica_count=1,\n",
" max_replica_count=1,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_proprietary_image_object_detection.ipynb\"\n",
" },\n",
")\n",
"\n",
"endpoint_id = endpoint.name\n",
notebooks/community/model_garden/model_garden_pytorch_autogluon.ipynb
+1
View File
@@ -305,6 +305,7 @@
" machine_type=machine_type,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_pytorch_autogluon.ipynb\"},\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_bart_large_cnn.ipynb
+1
View File
@@ -214,6 +214,7 @@
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_pytorch_bart_large_cnn.ipynb\"},\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_biogpt_serve.ipynb
+3
View File
@@ -299,6 +299,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_biogpt_serve.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_biomedclip.ipynb
+3
View File
@@ -317,6 +317,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_biomedclip.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_blip2.ipynb
+3
View File
@@ -292,6 +292,9 @@
" accelerator_type=\"NVIDIA_TESLA_T4\",\n",
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_blip2.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_blip_image_captioning.ipynb
+3
View File
@@ -291,6 +291,9 @@
" accelerator_type=\"NVIDIA_TESLA_T4\",\n",
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_blip_image_captioning.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_blip_vqa.ipynb
+1
View File
@@ -216,6 +216,7 @@
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_pytorch_blip_vqa.ipynb\"},\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_clip.ipynb
+3
View File
@@ -291,6 +291,9 @@
" accelerator_type=\"NVIDIA_TESLA_T4\",\n",
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_clip.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_controlnet.ipynb
+3
View File
@@ -221,6 +221,9 @@
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_controlnet.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_detectron2.ipynb
+3
View File
@@ -1177,6 +1177,9 @@
" min_replica_count=MIN_NODES,\n",
" max_replica_count=MAX_NODES,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_detectron2.ipynb\"\n",
" },\n",
")\n",
"\n",
"print(\"endpoint id is: \", endpoint.name)"
notebooks/community/model_garden/model_garden_pytorch_dolly_v2.ipynb
+3
View File
@@ -307,6 +307,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" # service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_dolly_v2.ipynb\"\n",
" },\n",
" )\n",
" print(\"endpoint_name:\", endpoint.name)\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_deployment.ipynb
+3
View File
@@ -207,6 +207,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_falcon_instruct_deployment.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_finetuning.ipynb
+3
View File
@@ -224,6 +224,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_falcon_instruct_finetuning.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_peft.ipynb
+3
View File
@@ -309,6 +309,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_falcon_instruct_peft.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_falcon_instruct_quantization.ipynb
+3
View File
@@ -221,6 +221,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_falcon_instruct_quantization.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_fill_mask.ipynb
+3
View File
@@ -185,6 +185,9 @@
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_fill_mask.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_flux.ipynb
+1
View File
@@ -267,6 +267,7 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_pytorch_flux.ipynb\"},\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_flux_gradio.ipynb
+1
View File
@@ -275,6 +275,7 @@
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" sync=False,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_pytorch_flux_gradio.ipynb\"},\n",
" )\n",
"\n",
" gr.Info(\n",
notebooks/community/model_garden/model_garden_pytorch_imagebind.ipynb
+3
View File
@@ -284,6 +284,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_imagebind.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_instant_id.ipynb
+1
View File
@@ -234,6 +234,7 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_pytorch_instant_id.ipynb\"},\n",
" )\n",
" print(\"To load this existing endpoint from a different session:\")\n",
" print(\n",
notebooks/community/model_garden/model_garden_pytorch_instant_id_gradio.ipynb
+3
View File
@@ -333,6 +333,9 @@
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" sync=False,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_instant_id_gradio.ipynb\"\n",
" },\n",
" )\n",
"\n",
" gr.Info(\n",
notebooks/community/model_garden/model_garden_pytorch_instructpix2pix.ipynb
+3
View File
@@ -215,6 +215,9 @@
" accelerator_type=\"NVIDIA_L4\",\n",
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_instructpix2pix.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_lama.ipynb
+3
View File
@@ -310,6 +310,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_lama.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_layoutml_document_qa.ipynb
+3
View File
@@ -291,6 +291,9 @@
" accelerator_type=\"NVIDIA_TESLA_T4\",\n",
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_layoutml_document_qa.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_llama2_deployment.ipynb
+3
View File
@@ -265,6 +265,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_llama2_deployment.ipynb\"\n",
" },\n",
" )\n",
" print(\"endpoint_name:\", endpoint.name)\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_llama2_peft_finetuning.ipynb
+3
View File
@@ -288,6 +288,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_llama2_peft_finetuning.ipynb\"\n",
" },\n",
" )\n",
" print(\"endpoint_name:\", endpoint.name)\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_llama2_peft_hyperparameter_tuning.ipynb
+3
View File
@@ -242,6 +242,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_llama2_peft_hyperparameter_tuning.ipynb\"\n",
" },\n",
" )\n",
" print(\"endpoint_name:\", endpoint.name)\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_llama2_quantization.ipynb
+3
View File
@@ -238,6 +238,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_llama2_quantization.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_llama3_1_qwen2_5_deployment_tpu.ipynb
+3
View File
@@ -381,6 +381,9 @@
" service_account=service_account,\n",
" min_replica_count=min_replica_count,\n",
" max_replica_count=max_replica_count,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_llama3_1_qwen2_5_deployment_tpu.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_nllb.ipynb
+3
View File
@@ -247,6 +247,9 @@
" accelerator_type=\"NVIDIA_TESLA_V100\",\n",
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=3600,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_nllb.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_open_clip.ipynb
+3
View File
@@ -373,6 +373,9 @@
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_open_clip.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_openllama_peft.ipynb
+3
View File
@@ -312,6 +312,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_openllama_peft.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_owlvit.ipynb
+1
View File
@@ -246,6 +246,7 @@
" accelerator_type=ACCELERATOR_TYPE,\n",
" accelerator_count=ACCELERATOR_COUNT,\n",
" deploy_request_timeout=1800,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_pytorch_owlvit.ipynb\"},\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_peft.ipynb
+3
View File
@@ -312,6 +312,9 @@
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_peft.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_pic2word.ipynb
+3
View File
@@ -295,6 +295,9 @@
" accelerator_count=1,\n",
" traffic_percentage=100,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_pic2word.ipynb\"\n",
" },\n",
")"
]
},
notebooks/community/model_garden/model_garden_pytorch_sam.ipynb
+3
View File
@@ -365,6 +365,9 @@
" accelerator_type=\"NVIDIA_TESLA_V100\",\n",
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_sam.ipynb\"\n",
" },\n",
" )\n",
" return endpoint, model"
]
notebooks/community/model_garden/model_garden_pytorch_sd_2_1_deployment.ipynb
+1
View File
@@ -230,6 +230,7 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_pytorch_sd_2_1_deployment.ipynb\"},\n",
" )\n",
" print(\"To load this existing endpoint from a different session:\")\n",
" print(\n",
notebooks/community/model_garden/model_garden_pytorch_sd_2_1_finetuning_dreambooth.ipynb
+3
View File
@@ -400,6 +400,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_sd_2_1_finetuning_dreambooth.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_sd_xl_finetuning_dreambooth_lora.ipynb
+3
View File
@@ -441,6 +441,9 @@
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_sd_xl_finetuning_dreambooth_lora.ipynb\"\n",
" },\n",
" )\n",
" print(\"To load this existing endpoint from a different session:\")\n",
" print(\n",
notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_custom.ipynb
+3
View File
@@ -242,6 +242,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_stable_diffusion_custom.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_deployment_1_5.ipynb
+3
View File
@@ -242,6 +242,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_stable_diffusion_deployment_1_5.ipynb\"\n",
" },\n",
" )\n",
" print(\"To load this existing endpoint from a different session:\")\n",
" print(\n",
notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_finetuning_dreambooth_1_5.ipynb
+3
View File
@@ -242,6 +242,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_stable_diffusion_finetuning_dreambooth_1_5.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_finetuning_dreambooth_lora_1_5.ipynb
+3
View File
@@ -344,6 +344,9 @@
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_stable_diffusion_finetuning_dreambooth_lora_1_5.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_finetuning_lora_1_5.ipynb
+3
View File
@@ -338,6 +338,9 @@
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_stable_diffusion_finetuning_lora_1_5.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_gradio.ipynb
+3
View File
@@ -487,6 +487,9 @@
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" sync=False,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_stable_diffusion_gradio.ipynb\"\n",
" },\n",
" )\n",
"\n",
" gr.Info(\n",
notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_inpainting.ipynb
+3
View File
@@ -218,6 +218,9 @@
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_stable_diffusion_inpainting.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_upscaler.ipynb
+3
View File
@@ -231,6 +231,9 @@
" accelerator_type=\"NVIDIA_L4\",\n",
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_stable_diffusion_upscaler.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_1_0.ipynb
+3
View File
@@ -232,6 +232,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_stable_diffusion_xl_1_0.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lcm.ipynb
+3
View File
@@ -238,6 +238,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_stable_diffusion_xl_lcm.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lightning.ipynb
+3
View File
@@ -218,6 +218,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_stable_diffusion_xl_lightning.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_lora.ipynb
+3
View File
@@ -234,6 +234,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_stable_diffusion_xl_lora.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_xl_turbo.ipynb
+3
View File
@@ -234,6 +234,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_stable_diffusion_xl_turbo.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_starcoder.ipynb
+3
View File
@@ -278,6 +278,9 @@
" accelerator_type=\"NVIDIA_TESLA_V100\",\n",
" accelerator_count=4,\n",
" deploy_request_timeout=3600,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_starcoder.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_text_to_video_zero_shot.ipynb
+3
View File
@@ -183,6 +183,9 @@
" accelerator_type=\"NVIDIA_L4\",\n",
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_text_to_video_zero_shot.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_timm.ipynb
+1
View File
@@ -561,6 +561,7 @@
" accelerator_count=1,\n",
" traffic_percentage=100,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_pytorch_timm.ipynb\"},\n",
")"
]
},
notebooks/community/model_garden/model_garden_pytorch_vicuna.ipynb
+3
View File
@@ -310,6 +310,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_vicuna.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_vilt_vqa.ipynb
+1
View File
@@ -215,6 +215,7 @@
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_pytorch_vilt_vqa.ipynb\"},\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_vit_gpt2_image_captioning.ipynb
+3
View File
@@ -291,6 +291,9 @@
" accelerator_type=\"NVIDIA_TESLA_T4\",\n",
" accelerator_count=1,\n",
" deploy_request_timeout=1800,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_vit_gpt2_image_captioning.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_whisper_large_v3_deployment.ipynb
+3
View File
@@ -243,6 +243,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_whisper_large_v3_deployment.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_pytorch_wizard_coder.ipynb
+3
View File
@@ -299,6 +299,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=1800,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_wizard_coder.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_pytorch_wizard_lm.ipynb
+3
View File
@@ -300,6 +300,9 @@
" accelerator_count=accelerator_count,\n",
" deploy_request_timeout=7200,\n",
" service_account=service_account,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_pytorch_wizard_lm.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint"
]
notebooks/community/model_garden/model_garden_recursion_mae_pytorch_inference_deployment.ipynb
+3
View File
@@ -179,6 +179,9 @@
" machine_type=MACHINE_TYPE,\n",
" deploy_request_timeout=1800,\n",
" service_account=\"\",\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_recursion_mae_pytorch_inference_deployment.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_tfvision_image_classification.ipynb
+1
View File
@@ -695,6 +695,7 @@
" accelerator_count=1,\n",
" min_replica_count=1,\n",
" max_replica_count=1,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_tfvision_image_classification.ipynb\"},\n",
")\n",
"\n",
"endpoint_id = endpoints[\"endpoint_icn\"].name\n",
notebooks/community/model_garden/model_garden_tfvision_image_object_detection.ipynb
+3
View File
@@ -939,6 +939,9 @@
" accelerator_count=1,\n",
" min_replica_count=1,\n",
" max_replica_count=1,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_tfvision_image_object_detection.ipynb\"\n",
" },\n",
")\n",
"\n",
"endpoint_id = endpoint.name\n",
notebooks/community/model_garden/model_garden_tfvision_image_segmentation.ipynb
+1
View File
@@ -678,6 +678,7 @@
" accelerator_count=1,\n",
" min_replica_count=1,\n",
" max_replica_count=1,\n",
" system_labels={\"NOTEBOOK_NAME\": \"model_garden_tfvision_image_segmentation.ipynb\"},\n",
")\n",
"\n",
"endpoint_id = endpoints[\"endpoint_isg\"].name\n",
notebooks/community/model_garden/model_garden_timesfm_deployment_on_vertex.ipynb
+3
View File
@@ -366,6 +366,9 @@
" enable_access_logging=True,\n",
" min_replica_count=1,\n",
" sync=True,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_timesfm_deployment_on_vertex.ipynb\"\n",
" },\n",
" )\n",
" return model, endpoint\n",
"\n",
notebooks/community/model_garden/model_garden_video_object_tracking_serve.ipynb
+3
View File
@@ -448,6 +448,9 @@
" accelerator_type=PREDICTION_ACCELERATOR_TYPE,\n",
" accelerator_count=1,\n",
" service_account=SERVICE_ACCOUNT,\n",
" system_labels={\n",
" \"NOTEBOOK_NAME\": \"model_garden_video_object_tracking_serve.ipynb\"\n",
" },\n",
" )\n",
" return endpoint, model"
]