Formatting and refactoring

PiperOrigin-RevId: 766506385

Author: vertex-mg-bot

notebooks/community/model_garden/model_garden_pytorch_sam.ipynb

@@ -116,6 +116,8 @@
         "! pip3 install --upgrade --quiet 'google-cloud-aiplatform==1.93.1'\n",
         "! git clone https://github.com/GoogleCloudPlatform/vertex-ai-samples.git\n",
         "\n",
+        "! pip install --quiet pycocotools\n",
+        "\n",
         "import importlib\n",
         "import os\n",
         "\n",
@@ -133,7 +135,6 @@
         ")\n",
         "\n",
         "models, endpoints = {}, {}\n",
-        "LABEL = \"sam_model\"\n",
         "\n",
         "\n",
         "# Get the default cloud project id.\n",
@@ -198,7 +199,8 @@
         "    task, display_name, model_id, machine_type, accelerator_type, accelerator_count\n",
         "):\n",
         "    endpoint = aiplatform.Endpoint.create(\n",
-        "        display_name=common_util.get_job_name_with_datetime(prefix=task)\n",
+        "        display_name=common_util.get_job_name_with_datetime(prefix=task),\n",
+        "        dedicated_endpoint_enabled=use_dedicated_endpoint,\n",
         "    )\n",
         "    serving_env = {\n",
         "        \"MODEL_ID\": model_id,\n",
@@ -239,6 +241,7 @@
       "source": [
         "# @title [Option 1] Deploy with Model Garden SDK\n",
         "\n",
+        "LABEL = \"sdk-deploy\"\n",
         "# @markdown Deploy with Gen AI model-centric SDK. This section uploads the prebuilt model to Model Registry and deploys it to a Vertex AI Endpoint. It takes 15 minutes to 1 hour to finish depending on the size of the model. See [use open models with Vertex AI](https://cloud.google.com/vertex-ai/generative-ai/docs/open-models/use-open-models) for documentation on other use cases.\n",
         "from vertexai.preview import model_garden\n",
         "\n",
@@ -277,14 +280,18 @@
         "    is_for_training=False,\n",
         ")\n",
         "\n",
-        "models[\"sam_model\"], endpoints[\"sam_endpoint\"] = deploy_model(\n",
+        "LABEL = \"custom-deploy\"\n",
+        "models[LABEL], endpoints[LABEL] = deploy_model(\n",
         "    task=task,\n",
         "    display_name=common_util.get_job_name_with_datetime(prefix=task),\n",
         "    model_id=\"facebook/sam-vit-large\",\n",
         "    machine_type=machine_type,\n",
         "    accelerator_type=accelerator_type,\n",
         "    accelerator_count=1,\n",
-        ")"
+        ")\n",
+        "\n",
+        "model = models[LABEL]\n",
+        "endpoint = endpoints[LABEL]"
       ]
     },
     {
@@ -347,7 +354,9 @@
         "    {\"image\": common_util.image_to_base64(image2)},\n",
         "]\n",
         "\n",
-        "preds = endpoints[\"sam_endpoint\"].predict(instances=instances).predictions\n",
+        "preds = endpoint.predict(\n",
+        "    instances=instances, use_dedicated_endpoint=use_dedicated_endpoint\n",
+        ").predictions\n",
         "show_predictions(preds)"
       ]
     },