fix: rename UJ10

pin kfp in google_cloud_pipeline_components_automl_text (#1748 )
pin kfp in google_cloud_pipeline_components_automl_images (#1746 )
2026-09-26 22:51:56 +00:00 · 2023-04-19 23:51:55 +00:00 · 2023-04-19 23:01:21 +00:00 · 2023-04-19 19:29:50 +00:00 · 2023-04-19 18:37:16 +00:00 · 2023-04-19 18:14:04 +00:00
57 changed files with 7637 additions and 441 deletions
@@ -59,7 +59,7 @@ managers: List[ResourceCleanupManager] = [
    TrainingJobCleanupManager(),
    HyperparameterTuningCleanupManager(),
    BatchPredictionJobCleanupManager(),
-    # ExperimentCleanupManager(), # Experiment missing _resource_noun
+    ExperimentCleanupManager(), # Experiment missing _resource_noun
    BucketCleanupManager()
 ]

@@ -131,6 +131,9 @@ class MatchingEngineIndexEndpointResourceCleanupManager(VertexAIResourceCleanupM
 class FeatureStoreCleanupManager(VertexAIResourceCleanupManager):
    vertex_ai_resource = aiplatform.Featurestore

+    def resource_name(self, resource: Any) -> str:
+        return resource.name
+
 class PipelineJobCleanupManager(VertexAIResourceCleanupManager):
    vertex_ai_resource = aiplatform.PipelineJob

@@ -166,6 +169,18 @@ class BatchPredictionJobCleanupManager(VertexAIResourceCleanupManager):
 class ExperimentCleanupManager(VertexAIResourceCleanupManager):
    vertex_ai_resource = aiplatform.Experiment

+    @property
+    def type_name(self) -> str:
+        return "Experiment"
+
+    def resource_name(self, resource: Any) -> str:
+        return resource.name
+
+    def get_seconds_since_modification(self, resource: Any) -> float:
+        update_time = resource._metadata_context.update_time
+        current_time = DatetimeWithNanoseconds.now()
+        return float(current_time.timestamp() - update_time.timestamp())
+
 class BucketCleanupManager(ResourceCleanupManager):
    vertex_ai_resource = storage.bucket.Bucket

@@ -181,7 +196,7 @@ class BucketCleanupManager(ResourceCleanupManager):

    @property
    def type_name(self) -> str:
-        return str(type(self.vertex_ai_resource))
+        return "Bucket"

    def get_seconds_since_modification(self, resource: Any) -> float:
        # Bucket has no last_update property, only time created
@@ -195,7 +210,7 @@ class BucketCleanupManager(ResourceCleanupManager):
    def is_deletable(self, resource: Any) -> bool:
        time_difference = self.get_seconds_since_modification(resource)

-        if self.resource_name(resource).startswith('your-bucket-name'):
+        if not self.resource_name(resource).startswith('your-bucket-name'):
            print(f"Skipping '{resource}' not a Vertex AI notebook bucket")
            return False

@@ -55,3 +55,4 @@
 /notebooks/community/model_garden/model_garden_pytorch_clip.ipynb @xiangxu-google
 /notebooks/community/model_garden/model_garden_pytorch_owlvit.ipynb @xiangxu-google
 /notebooks/community/model_garden/model_garden_pytorch_layoutml_document_qa.ipynb @xiangxu-google
+/notebooks/community/model_garden/model_garden_pytorch_blip2.ipynb @xiangxu-google
@@ -29,22 +29,22 @@
        "id": "title:generic,gcp"
      },
      "source": [
-        "# E2E ML on GCP: MLOps stage 4 : formalization: get started with Vertex ML Metadata\n",
+        "# Get started with Vertex ML Metadata\n",
        "\n",
        "<table align=\"left\">\n",
        "  <td>\n",
-        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/ml_ops/stage4/get_started_with_vertex_ml_metadata.ipynb\">\n",
+        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/ml_metadata/get_started_with_vertex_ml_metadata.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
        "  <td>\n",
-        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/ml_ops/stage4/get_started_with_vertex_ml_metadata.ipynb\">\n",
+        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/ml_metadata/get_started_with_vertex_ml_metadata.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
        "  </td>\n",
        "  <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/ml_ops/stage4/get_started_with_vertex_ml_metadata.ipynb\">\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/official/ml_metadata/get_started_with_vertex_ml_metadata.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "      Open in Vertex AI Workbench\n",
        "    </a>\n",
@@ -62,7 +62,9 @@
        "## Overview\n",
        "\n",
        "\n",
-        "This tutorial demonstrates how to use Vertex AI for E2E MLOps on Google Cloud in production. This tutorial covers stage 4 : formalization: get started with Vertex ML Metadata."
+        "This tutorial demonstrates how to use Vertex ML Metadata.\n",
+        "\n",
+        "Learn more about [Vertex ML Metadata](https://cloud.google.com/vertex-ai/docs/ml-metadata)."
      ]
    },
    {
@@ -144,19 +146,32 @@
      "source": [
        "import os\n",
        "\n",
-        "# The Vertex AI Workbench Notebook product has specific requirements\n",
-        "IS_WORKBENCH_NOTEBOOK = os.getenv(\"DL_ANACONDA_HOME\")\n",
-        "IS_USER_MANAGED_WORKBENCH_NOTEBOOK = os.path.exists(\n",
-        "    \"/opt/deeplearning/metadata/env_version\"\n",
-        ")\n",
+        "! pip3 install --upgrade google-cloud-aiplatform[tensorboard] \\\n",
+        "                         google-cloud-pipeline-components  --quiet"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "restart"
+      },
+      "source": [
+        "### Colab only: Uncomment the following cell to restart the kernel"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "D-ZBOjErv5mM"
+      },
+      "outputs": [],
+      "source": [
+        "# Automatically restart kernel after installs so that your environment can access the new packages\n",
+        "# import IPython\n",
        "\n",
-        "# Vertex AI Notebook requires dependencies to be installed with '--user'\n",
-        "USER_FLAG = \"\"\n",
-        "if IS_WORKBENCH_NOTEBOOK:\n",
-        "    USER_FLAG = \"--user\"\n",
-        "\n",
-        "! pip3 install --upgrade google-cloud-aiplatform[tensorboard] $USER_FLAG -q\n",
-        "! pip3 install --upgrade google-cloud-pipeline-components $USER_FLAG -q"
+        "# app = IPython.Application.instance()\n",
+        "# app.kernel.do_shutdown(True)"
      ]
    },
    {
@@ -170,63 +185,27 @@
        "Once you've installed the additional packages, you need to restart the notebook kernel so it can find the packages."
      ]
    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "restart"
-      },
-      "outputs": [],
-      "source": [
-        "import os\n",
-        "\n",
-        "if not os.getenv(\"IS_TESTING\"):\n",
-        "    # Automatically restart kernel after installs\n",
-        "    import IPython\n",
-        "\n",
-        "    app = IPython.Application.instance()\n",
-        "    app.kernel.do_shutdown(True)"
-      ]
-    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "before_you_begin"
      },
      "source": [
-        "## Before you begin\n",
-        "\n",
-        "### GPU runtime\n",
-        "\n",
-        "*Make sure you're running this notebook in a GPU runtime if you have that option. In Colab, select* **Runtime > Change Runtime Type > GPU**\n",
-        "\n",
-        "### Set up your Google Cloud project\n",
-        "\n",
-        "**The following steps are required, regardless of your notebook environment.**\n",
-        "\n",
-        "1. [Select or create a Google Cloud project](https://console.cloud.google.com/cloud-resource-manager). When you first create an account, you get a $300 free credit towards your compute/storage costs.\n",
-        "\n",
-        "2. [Make sure that billing is enabled for your project.](https://cloud.google.com/billing/docs/how-to/modify-project)\n",
-        "\n",
-        "3. [Enable the following APIs: Vertex AI APIs, Compute Engine APIs, and Cloud Storage.](https://console.cloud.google.com/flows/enableapi?apiid=aiplatform.googleapis.com,compute_component,storage-component.googleapis.com)\n",
-        "\n",
-        "4. If you are running this notebook locally, you need to install the [Cloud SDK]((https://cloud.google.com/sdk)).\n",
-        "\n",
-        "5. Enter your project ID in the cell below. Then run the  cell to make sure the\n",
-        "Cloud SDK uses the right project for all the commands in this notebook.\n",
-        "\n",
-        "**Note**: Jupyter runs lines prefixed with `!` as shell commands, and it interpolates Python variables prefixed with `$`."
+        "## Before you begin"
      ]
    },
    {
      "cell_type": "markdown",
      "metadata": {
-        "id": "project_id"
+        "id": "before_you_begin:nogpu"
      },
      "source": [
-        "#### Set your project ID\n",
+        "### Set your project ID\n",
        "\n",
-        "**If you don't know your project ID**, you may be able to get your project ID using `gcloud`."
+        "**If you don't know your project ID**, try the following:\n",
+        "* Run `gcloud config list`.\n",
+        "* Run `gcloud projects list`.\n",
+        "* See the support page: [Locate the project ID](https://support.google.com/googleapi/answer/7014113)"
      ]
    },
    {
@@ -237,33 +216,10 @@
      },
      "outputs": [],
      "source": [
-        "PROJECT_ID = \"[your-project-id]\"  # @param {type:\"string\"}"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "autoset_project_id"
-      },
-      "outputs": [],
-      "source": [
-        "if PROJECT_ID == \"\" or PROJECT_ID is None or PROJECT_ID == \"[your-project-id]\":\n",
-        "    # Get your GCP project id from gcloud\n",
-        "    shell_output = ! gcloud config list --format 'value(core.project)' 2>/dev/null\n",
-        "    PROJECT_ID = shell_output[0]\n",
-        "    print(\"Project ID:\", PROJECT_ID)"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "set_gcloud_project_id"
-      },
-      "outputs": [],
-      "source": [
-        "! gcloud config set project $PROJECT_ID"
+        "PROJECT_ID = \"[your-project-id]\"  # @param {type:\"string\"}\n",
+        "\n",
+        "# Set the project id\n",
+        "! gcloud config set project {PROJECT_ID}"
      ]
    },
    {
@@ -274,16 +230,7 @@
      "source": [
        "#### Region\n",
        "\n",
-        "You can also change the `REGION` variable, which is used for operations\n",
-        "throughout the rest of this notebook.  Below are regions supported for Vertex AI. We recommend that you choose the region closest to you.\n",
-        "\n",
-        "- Americas: `us-central1`\n",
-        "- Europe: `europe-west4`\n",
-        "- Asia Pacific: `asia-east1`\n",
-        "\n",
-        "You may not use a multi-regional bucket for training with Vertex AI. Not all regions provide support for all Vertex AI services.\n",
-        "\n",
-        "Learn more about [Vertex AI regions](https://cloud.google.com/vertex-ai/docs/general/locations)."
+        "You can also change the `REGION` variable used by Vertex AI. Learn more about [Vertex AI regions](https://cloud.google.com/vertex-ai/docs/general/locations)."
      ]
    },
    {
@@ -294,34 +241,7 @@
      },
      "outputs": [],
      "source": [
-        "REGION = \"[your-region]\"  # @param {type: \"string\"}\n",
-        "\n",
-        "if REGION == \"[your-region]\":\n",
-        "    REGION = \"us-central1\""
-      ]
-    },
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "timestamp"
-      },
-      "source": [
-        "#### Timestamp\n",
-        "\n",
-        "If you are in a live tutorial session, you might be using a shared test account or project. To avoid name collisions between users on resources created, you create a timestamp for each instance session, and append the timestamp onto the name of resources you create in this tutorial."
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "timestamp"
-      },
-      "outputs": [],
-      "source": [
-        "from datetime import datetime\n",
-        "\n",
-        "TIMESTAMP = datetime.now().strftime(\"%Y%m%d%H%M%S\")"
+        "REGION = \"us-central1\"  # @param {type: \"string\"}"
      ]
    },
    {
@@ -332,57 +252,70 @@
      "source": [
        "### Authenticate your Google Cloud account\n",
        "\n",
-        "**If you are using Vertex AI Workbench Notebooks**, your environment is already authenticated. Skip this step.\n",
-        "\n",
-        "**If you are using Colab**, run the cell below and follow the instructions when prompted to authenticate your account via oAuth.\n",
-        "\n",
-        "**Otherwise**, follow these steps:\n",
-        "\n",
-        "In the Cloud Console, go to the [Create service account key](https://console.cloud.google.com/apis/credentials/serviceaccountkey) page.\n",
-        "\n",
-        "**Click Create service account**.\n",
-        "\n",
-        "In the **Service account name** field, enter a name, and click **Create**.\n",
-        "\n",
-        "In the **Grant this service account access to project** section, click the Role drop-down list. Type \"Vertex\" into the filter box, and select **Vertex Administrator**. Type \"Storage Object Admin\" into the filter box, and select **Storage Object Admin**.\n",
-        "\n",
-        "Click Create. A JSON file that contains your key downloads to your local environment.\n",
-        "\n",
-        "Enter the path to your service account key as the GOOGLE_APPLICATION_CREDENTIALS variable in the cell below and run the cell."
+        "Depending on your Jupyter environment, you may have to manually authenticate. Follow the relevant instructions below."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "FvQeFm3Gv5mR"
+      },
+      "source": [
+        "**1. Vertex AI Workbench**\n",
+        "* Do nothing as you are already authenticated."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "ad1138a125ea"
+      },
+      "source": [
+        "**2. Local JupyterLab instance, uncomment and run:**"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
-        "id": "gcp_authenticate"
+        "id": "ce6043da7b33"
+      },
+      "outputs": [],
+      "source": [
+        "# ! gcloud auth login"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "0367eac06a10"
+      },
+      "source": [
+        "**3. Colab, uncomment and run:**"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "21ad4dbb4a61"
      },
      "outputs": [],
      "source": [
-        "# If you are running this notebook in Colab, run this cell and follow the\n",
-        "# instructions to authenticate your GCP account. This provides access to your\n",
-        "# Cloud Storage bucket and lets you submit training jobs and prediction\n",
-        "# requests.\n",
-        "\n",
-        "import os\n",
-        "import sys\n",
-        "\n",
-        "# If on Vertex AI Workbench, then don't execute this code\n",
        "IS_COLAB = False\n",
-        "if not os.path.exists(\"/opt/deeplearning/metadata/env_version\") and not os.getenv(\n",
-        "    \"DL_ANACONDA_HOME\"\n",
-        "):\n",
-        "    if \"google.colab\" in sys.modules:\n",
-        "        IS_COLAB = True\n",
-        "        from google.colab import auth as google_auth\n",
-        "\n",
-        "        google_auth.authenticate_user()\n",
-        "\n",
-        "    # If you are running this notebook locally, replace the string below with the\n",
-        "    # path to your service account key and run this cell to authenticate your GCP\n",
-        "    # account.\n",
-        "    elif not os.getenv(\"IS_TESTING\"):\n",
-        "        %env GOOGLE_APPLICATION_CREDENTIALS ''"
+        "# from google.colab import auth\n",
+        "# auth.authenticate_user()\n",
+        "# IS_COLAB = True"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "c13224697bfb"
+      },
+      "source": [
+        "**4. Service account or other**\n",
+        "* See how to grant Cloud Storage permissions to your service account at https://cloud.google.com/storage/docs/gsutil/commands/iam#ch-examples."
      ]
    },
    {
@@ -393,11 +326,7 @@
      "source": [
        "### Create a Cloud Storage bucket\n",
        "\n",
-        "**The following steps are required, regardless of your notebook environment.**\n",
-        "\n",
-        "When you initialize the Vertex SDK for Python, you specify a Cloud Storage staging bucket. The staging bucket is where all the data associated with your dataset and model resources are retained across sessions.\n",
-        "\n",
-        "Set the name of your Cloud Storage bucket below. Bucket names must be globally unique across all Google Cloud projects, including those outside of your organization."
+        "Create a storage bucket to store intermediate artifacts such as datasets."
      ]
    },
    {
@@ -408,21 +337,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_NAME = \"[your-bucket-name]\"  # @param {type:\"string\"}\n",
-        "BUCKET_URI = f\"gs://{BUCKET_NAME}\""
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "autoset_bucket"
-      },
-      "outputs": [],
-      "source": [
-        "if BUCKET_NAME == \"\" or BUCKET_NAME is None or BUCKET_NAME == \"[your-bucket-name]\":\n",
-        "    BUCKET_NAME = PROJECT_ID + \"aip-\" + TIMESTAMP\n",
-        "    BUCKET_URI = \"gs://\" + BUCKET_NAME"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -445,26 +360,6 @@
        "! gsutil mb -l $REGION $BUCKET_URI"
      ]
    },
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "validate_bucket"
-      },
-      "source": [
-        "Finally, validate access to your Cloud Storage bucket by examining its contents:"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "validate_bucket"
-      },
-      "outputs": [],
-      "source": [
-        "! gsutil ls -al $BUCKET_URI"
-      ]
-    },
    {
      "cell_type": "markdown",
      "metadata": {
@@ -545,19 +440,7 @@
      "source": [
        "### Set up variables\n",
        "\n",
-        "Next, set up some variables used throughout the tutorial.\n",
-        "### Import libraries and define constants"
-      ]
-    },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "id": "import_aip:mbsdk"
-      },
-      "outputs": [],
-      "source": [
-        "import google.cloud.aiplatform as aip"
+        "Next, set up some variables used throughout the tutorial."
      ]
    },
    {
@@ -579,7 +462,10 @@
      },
      "outputs": [],
      "source": [
-        "import google.cloud.aiplatform_v1beta1 as aip_beta"
+        "from datetime import datetime\n",
+        "\n",
+        "import google.cloud.aiplatform_v1beta1 as aip_beta\n",
+        "from google.cloud import aiplatform"
      ]
    },
    {
@@ -707,7 +593,7 @@
      "outputs": [],
      "source": [
        "metadata_store = clients[\"metadata\"].create_metadata_store(\n",
-        "    parent=PARENT, metadata_store_id=\"my-metadata-store\"\n",
+        "    parent=PARENT, metadata_store_id=\"my-metadata-store-unique\"\n",
        ")\n",
        "\n",
        "metadata_store_id = str(metadata_store.result())[7:-2]\n",
@@ -1144,7 +1030,7 @@
      "source": [
        "from kfp.v2 import compiler, dsl\n",
        "from kfp.v2.dsl import (Artifact, Dataset, Input, Metrics, Model, Output,\n",
-        "                        OutputPath, component, pipeline)"
+        "                        OutputPath, component)"
      ]
    },
    {
@@ -1196,7 +1082,7 @@
      "outputs": [],
      "source": [
        "@component(\n",
-        "    packages_to_install=[\"google-cloud-bigquery\", \"pandas\", \"pyarrow\"],\n",
+        "    packages_to_install=[\"google-cloud-bigquery\", \"pandas\", \"pyarrow\", \"db-dtypes\"],\n",
        "    base_image=\"python:3.9\",\n",
        "    output_component_file=\"create_dataset.yaml\",\n",
        ")\n",
@@ -1214,7 +1100,7 @@
        "\n",
        "\n",
        "@component(\n",
-        "    packages_to_install=[\"sklearn\", \"pandas\", \"joblib\"],\n",
+        "    packages_to_install=[\"scikit-learn\", \"pandas\", \"joblib\"],\n",
        "    base_image=\"python:3.9\",\n",
        "    output_component_file=\"beans_model_component.yaml\",\n",
        ")\n",
@@ -1298,7 +1184,7 @@
        "    # A name for the pipeline.\n",
        "    name=\"mlmd-pipeline\",\n",
        ")\n",
-        "def pipeline(\n",
+        "def my_pipeline(\n",
        "    bq_table: str = \"\",\n",
        "    output_data_path: str = \"data.csv\",\n",
        "    project: str = PROJECT_ID,\n",
@@ -1332,20 +1218,22 @@
      "source": [
        "NOW = datetime.now().isoformat().replace(\".\", \":\")[:-7]\n",
        "\n",
-        "compiler.Compiler().compile(pipeline_func=pipeline, package_path=\"mlmd_pipeline.json\")\n",
+        "compiler.Compiler().compile(\n",
+        "    pipeline_func=my_pipeline, package_path=\"mlmd_pipeline.json\"\n",
+        ")\n",
        "\n",
-        "run1 = aip.PipelineJob(\n",
+        "run1 = aiplatform.PipelineJob(\n",
        "    display_name=\"mlmd-pipeline\",\n",
        "    template_path=\"mlmd_pipeline.json\",\n",
-        "    job_id=\"mlmd-pipeline-small-{}\".format(TIMESTAMP),\n",
+        "    job_id=\"mlmd-pipeline-small-unique\",\n",
        "    parameter_values={\"bq_table\": \"sara-vertex-demos.beans_demo.small_dataset\"},\n",
        "    enable_caching=True,\n",
        ")\n",
        "\n",
-        "run2 = aip.PipelineJob(\n",
+        "run2 = aiplatform.PipelineJob(\n",
        "    display_name=\"mlmd-pipeline\",\n",
        "    template_path=\"mlmd_pipeline.json\",\n",
-        "    job_id=\"mlmd-pipeline-large-{}\".format(TIMESTAMP),\n",
+        "    job_id=\"mlmd-pipeline-large-unique\",\n",
        "    parameter_values={\"bq_table\": \"sara-vertex-demos.beans_demo.large_dataset\"},\n",
        "    enable_caching=True,\n",
        ")\n",
@@ -1382,7 +1270,7 @@
      },
      "outputs": [],
      "source": [
-        "df = aip.get_pipeline_df(pipeline=\"mlmd-pipeline\")\n",
+        "df = aiplatform.get_pipeline_df(pipeline=\"mlmd-pipeline\")\n",
        "print(df)"
      ]
    },
@@ -1466,6 +1354,10 @@
      },
      "outputs": [],
      "source": [
+        "metadata_store_id = (\n",
+        "    f\"projects/{PROJECT_ID}/locations/{REGION}/metadataStores/my-metadata-store-unique\"\n",
+        ")\n",
+        "\n",
        "clients[\"metadata\"].delete_metadata_store(name=metadata_store_id)"
      ]
    },
@@ -0,0 +1,468 @@
+{
+  "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "7d9bbf86da5e"
+      },
+      "outputs": [],
+      "source": [
+        "# Copyright 2023 Google LLC\n",
+        "#\n",
+        "# Licensed under the Apache License, Version 2.0 (the \"License\");\n",
+        "# you may not use this file except in compliance with the License.\n",
+        "# You may obtain a copy of the License at\n",
+        "#\n",
+        "#     https://www.apache.org/licenses/LICENSE-2.0\n",
+        "#\n",
+        "# Unless required by applicable law or agreed to in writing, software\n",
+        "# distributed under the License is distributed on an \"AS IS\" BASIS,\n",
+        "# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n",
+        "# See the License for the specific language governing permissions and\n",
+        "# limitations under the License."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "2bd716bf3e39"
+      },
+      "source": [
+        "# Vertex AI Model Garden - BLIP2\n",
+        "\n",
+        "<table align=\"left\">\n",
+        "  <td>\n",
+        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_blip2.ipynb\">\n",
+        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
+        "    </a>\n",
+        "  </td>\n",
+        "  <td>\n",
+        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_blip2.ipynb\">\n",
+        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
+        "      View on GitHub\n",
+        "    </a>\n",
+        "  </td>\n",
+        "  <td>\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/model_garden/model_garden_pytorch_blip2.ipynb\">\n",
+        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
+        "Open in Vertex AI Workbench\n",
+        "    </a>\n",
+        "    (a Python-3 CPU notebook is recommended)\n",
+        "  </td>\n",
+        "</table>"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "d8cd12648da4"
+      },
+      "source": [
+        "## Overview\n",
+        "\n",
+        "This notebook demonstrates deploying the pre-trained [BLIP2](https://huggingface.co/Salesforce/blip2-opt-2.7b) model on Vertex AI for online prediction.\n",
+        "\n",
+        "### Objective\n",
+        "\n",
+        "- Upload the model to [Model Registry](https://cloud.google.com/vertex-ai/docs/model-registry/introduction).\n",
+        "- Deploy the model on [Endpoint](https://cloud.google.com/vertex-ai/docs/predictions/using-private-endpoints).\n",
+        "- Run online predictions for image captioning.\n",
+        "\n",
+        "### Costs\n",
+        "\n",
+        "This tutorial uses billable components of Google Cloud:\n",
+        "\n",
+        "* Vertex AI\n",
+        "* Cloud Storage\n",
+        "\n",
+        "Learn about [Vertex AI pricing](https://cloud.google.com/vertex-ai/pricing) and [Cloud Storage pricing](https://cloud.google.com/storage/pricing), and use the [Pricing Calculator](https://cloud.google.com/products/calculator/) to generate a cost estimate based on your projected usage."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "264c07757582"
+      },
+      "source": [
+        "## Setup environment\n",
+        "\n",
+        "**NOTE**: Jupyter runs lines prefixed with `!` as shell commands, and it interpolates Python variables prefixed with `$` into these commands."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "d73ffa0c0b83"
+      },
+      "source": [
+        "### Colab only"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "2707b02ef5df"
+      },
+      "outputs": [],
+      "source": [
+        "!pip3 install --upgrade google-cloud-aiplatform"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "b60a4d7100bf"
+      },
+      "outputs": [],
+      "source": [
+        "from google.colab import auth as google_auth\n",
+        "\n",
+        "google_auth.authenticate_user()"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "0f826ff482a2"
+      },
+      "source": [
+        "### Setup Google Cloud project\n",
+        "\n",
+        "1. [Select or create a Google Cloud project](https://console.cloud.google.com/cloud-resource-manager). When you first create an account, you get a $300 free credit towards your compute/storage costs.\n",
+        "\n",
+        "1. [Make sure that billing is enabled for your project](https://cloud.google.com/billing/docs/how-to/modify-project).\n",
+        "\n",
+        "1. [Enable the Vertex AI API and Compute Engine API](https://console.cloud.google.com/flows/enableapi?apiid=aiplatform.googleapis.com,compute_component).\n",
+        "\n",
+        "1. [Create a Cloud Storage bucket](https://cloud.google.com/storage/docs/creating-buckets) for storing experiment outputs."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "8958ebc71868"
+      },
+      "source": [
+        "Fill following variables for experiments environment:"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "9db30f827a65"
+      },
+      "outputs": [],
+      "source": [
+        "# Cloud project id.\n",
+        "PROJECT_ID = \"\"  # @param {type:\"string\"}\n",
+        "\n",
+        "# The region you want to launch jobs in.\n",
+        "REGION = \"us-central1\"  # @param {type:\"string\"}\n",
+        "\n",
+        "# The Cloud Storage bucket for storing experiments output. Fill it without the 'gs://' prefix.\n",
+        "GCS_BUCKET = \"\"  # @param {type:\"string\"}"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "92f16e22c20b"
+      },
+      "source": [
+        "Initialize Vertex AI API:"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "1680c257acfb"
+      },
+      "outputs": [],
+      "source": [
+        "from google.cloud import aiplatform\n",
+        "\n",
+        "aiplatform.init(project=PROJECT_ID, location=REGION, staging_bucket=GCS_BUCKET)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "6ca48b699d17"
+      },
+      "source": [
+        "### Define constants"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "de9882ea89ea"
+      },
+      "outputs": [],
+      "source": [
+        "# The pre-built serving docker image.\n",
+        "# The model artifacts are embedded within the container, except for model weights which will be downloaded during deployment.\n",
+        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve\""
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "10188266a5cd"
+      },
+      "source": [
+        "### Define common functions"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "cac4478ae098"
+      },
+      "outputs": [],
+      "source": [
+        "import base64\n",
+        "import os\n",
+        "from datetime import datetime\n",
+        "from io import BytesIO\n",
+        "\n",
+        "import requests\n",
+        "from google.cloud import aiplatform\n",
+        "from PIL import Image\n",
+        "\n",
+        "\n",
+        "def create_job_name(prefix):\n",
+        "    user = os.environ.get(\"USER\")\n",
+        "    now = datetime.now().strftime(\"%Y%m%d_%H%M%S\")\n",
+        "    job_name = f\"{prefix}-{user}-{now}\"\n",
+        "    return job_name\n",
+        "\n",
+        "\n",
+        "def download_image(url):\n",
+        "    response = requests.get(url)\n",
+        "    return Image.open(BytesIO(response.content))\n",
+        "\n",
+        "\n",
+        "def image_to_base64(image, format=\"JPEG\"):\n",
+        "    buffer = BytesIO()\n",
+        "    image.save(buffer, format=format)\n",
+        "    image_str = base64.b64encode(buffer.getvalue()).decode(\"utf-8\")\n",
+        "    return image_str\n",
+        "\n",
+        "\n",
+        "def base64_to_image(image_str):\n",
+        "    image = Image.open(BytesIO(base64.b64decode(image_str)))\n",
+        "    return image\n",
+        "\n",
+        "\n",
+        "def image_grid(imgs, rows=2, cols=2):\n",
+        "    w, h = imgs[0].size\n",
+        "    grid = Image.new(\"RGB\", size=(cols * w, rows * h))\n",
+        "    for i, img in enumerate(imgs):\n",
+        "        grid.paste(img, box=(i % cols * w, i // cols * h))\n",
+        "    return grid\n",
+        "\n",
+        "\n",
+        "def deploy_model(model_id, task):\n",
+        "    model_name = \"blip-image-captioning\"\n",
+        "    endpoint = aiplatform.Endpoint.create(display_name=f\"{model_name}-endpoint\")\n",
+        "    serving_env = {\n",
+        "        \"MODEL_ID\": model_id,\n",
+        "        \"TASK\": task,\n",
+        "    }\n",
+        "    # If the model_id is a GCS path, use artifact_uri to pass it to serving docker.\n",
+        "    artifact_uri = model_id if model_id.startswith(\"gs://\") else None\n",
+        "    model = aiplatform.Model.upload(\n",
+        "        display_name=model_name,\n",
+        "        serving_container_image_uri=SERVE_DOCKER_URI,\n",
+        "        serving_container_ports=[7080],\n",
+        "        serving_container_predict_route=\"/predictions/transformers_serving\",\n",
+        "        serving_container_health_route=\"/ping\",\n",
+        "        serving_container_environment_variables=serving_env,\n",
+        "        artifact_uri=artifact_uri,\n",
+        "    )\n",
+        "    model.deploy(\n",
+        "        endpoint=endpoint,\n",
+        "        machine_type=\"n1-standard-8\",\n",
+        "        accelerator_type=\"NVIDIA_TESLA_T4\",\n",
+        "        accelerator_count=1,\n",
+        "        deploy_request_timeout=1800,\n",
+        "    )\n",
+        "    return model, endpoint"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "d2d72ecdb8c9"
+      },
+      "source": [
+        "## Upload and deploy models"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "9448c5f545fa"
+      },
+      "source": [
+        "This section uploads the pre-trained model to Model Registry and deploys it on the Endpoint with 1 T4 GPU.\n",
+        "\n",
+        "The model deployment step will take ~15 minutes to complete.\n",
+        "\n",
+        "Once deployed, you can send images to get descriptions."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "19e8aeec969c"
+      },
+      "source": [
+        "### Image captioning"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "b4b46c28d8b1"
+      },
+      "outputs": [],
+      "source": [
+        "model, endpoint = deploy_model(\n",
+        "    model_id=\"Salesforce/blip2-opt-2.7b\", task=\"image-to-text\"\n",
+        ")"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "12893aa2c5af"
+      },
+      "source": [
+        "NOTE: The model weights will be downloaded after the deployment succeeds. When the model is very large it could add 5~15mins additional time before the endpoint is ready for prediction."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "6be655247cb1"
+      },
+      "outputs": [],
+      "source": [
+        "image = download_image(\"http://images.cocodataset.org/val2017/000000039769.jpg\")\n",
+        "display(image)\n",
+        "\n",
+        "instances = [\n",
+        "    {\"image\": image_to_base64(image)},\n",
+        "]\n",
+        "preds = endpoint.predict(instances=instances).predictions\n",
+        "print(preds)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "2ccf3714dbe9"
+      },
+      "outputs": [],
+      "source": [
+        "# Undeploy model and delete endpoint.\n",
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "01e11d85d504"
+      },
+      "source": [
+        "### VQA (Visual-Question-Answering)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "26018d961cf9"
+      },
+      "outputs": [],
+      "source": [
+        "model, endpoint = deploy_model(\n",
+        "    model_id=\"Salesforce/blip2-opt-2.7b\", task=\"visual-question-answering\"\n",
+        ")"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "0ac7f8d945e3"
+      },
+      "source": [
+        "NOTE: The model weights will be downloaded after the deployment succeeds. When the model is very large it could add 5~15mins additional time before the endpoint is ready for prediction."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "f19c342829fd"
+      },
+      "outputs": [],
+      "source": [
+        "image = download_image(\n",
+        "    \"https://media.newyorker.com/cartoons/63dc6847be24a6a76d90eb99/master/w_1160,c_limit/230213_a26611_838.jpg\"\n",
+        ")\n",
+        "display(image)\n",
+        "\n",
+        "question = \"Question: What are they doing? Answer:\"\n",
+        "instances = [\n",
+        "    {\"image\": image_to_base64(image), \"text\": question},\n",
+        "]\n",
+        "preds = endpoint.predict(instances=instances).predictions\n",
+        "print(question)\n",
+        "print(preds)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "712eb9d0b336"
+      },
+      "outputs": [],
+      "source": [
+        "# Undeploy model and delete endpoint.\n",
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
+      ]
+    }
+  ],
+  "metadata": {
+    "colab": {
+      "name": "model_garden_pytorch_blip2.ipynb",
+      "toc_visible": true
+    },
+    "kernelspec": {
+      "display_name": "Python 3",
+      "name": "python3"
+    }
+  },
+  "nbformat": 4,
+  "nbformat_minor": 0
+}
@@ -37,18 +37,18 @@
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
-        "\n",
        "  <td>\n",
        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_blip_image_captioning.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
        "  </td>\n",
-        "  <td>                                                                                               <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_blip_image_captioning.ipynb\">\n",
+        "  <td>\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/model_garden/model_garden_pytorch_blip_image_captioning.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "Open in Vertex AI Workbench\n",
        "    </a>\n",
+        "    (a Python-3 CPU notebook is recommended)\n",
        "  </td>\n",
        "</table>"
      ]
@@ -207,7 +207,7 @@
      "outputs": [],
      "source": [
        "# The pre-built serving docker image. It contains serving scripts and models.\n",
-        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve:latest\""
+        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve\""
      ]
    },
    {
@@ -282,7 +282,7 @@
        "        display_name=model_name,\n",
        "        serving_container_image_uri=SERVE_DOCKER_URI,\n",
        "        serving_container_ports=[7080],\n",
-        "        serving_container_predict_route=\"/predictions/diffusers_serving\",\n",
+        "        serving_container_predict_route=\"/predictions/transformers_serving\",\n",
        "        serving_container_health_route=\"/ping\",\n",
        "        serving_container_environment_variables=serving_env,\n",
        "        artifact_uri=artifact_uri,\n",
@@ -367,11 +367,11 @@
      },
      "outputs": [],
      "source": [
-        "# Delete models.\n",
-        "model.delete()\n",
-        "\n",
        "# Undeploy model and delete endpoint.\n",
-        "endpoint.delete(force=True)"
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
      ]
    }
  ],
@@ -37,18 +37,18 @@
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
-        "\n",
        "  <td>\n",
        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_blip_vqa.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
        "  </td>\n",
-        "  <td>                                                                                               <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_blip_vqa.ipynb\">\n",
+        "  <td>\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/model_garden/model_garden_pytorch_blip_vqa.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "Open in Vertex AI Workbench\n",
        "    </a>\n",
+        "    (a Python-3 CPU notebook is recommended)\n",
        "  </td>\n",
        "</table>"
      ]
@@ -207,7 +207,7 @@
      "outputs": [],
      "source": [
        "# The pre-built serving docker image. It contains serving scripts and models.\n",
-        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve:latest\""
+        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve\""
      ]
    },
    {
@@ -282,7 +282,7 @@
        "        display_name=model_name,\n",
        "        serving_container_image_uri=SERVE_DOCKER_URI,\n",
        "        serving_container_ports=[7080],\n",
-        "        serving_container_predict_route=\"/predictions/diffusers_serving\",\n",
+        "        serving_container_predict_route=\"/predictions/transformers_serving\",\n",
        "        serving_container_health_route=\"/ping\",\n",
        "        serving_container_environment_variables=serving_env,\n",
        "        artifact_uri=artifact_uri,\n",
@@ -369,11 +369,11 @@
      },
      "outputs": [],
      "source": [
-        "# Delete models.\n",
-        "model.delete()\n",
-        "\n",
        "# Undeploy model and delete endpoint.\n",
-        "endpoint.delete(force=True)"
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
      ]
    }
  ],
@@ -37,18 +37,18 @@
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
-        "\n",
        "  <td>\n",
        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_clip.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
        "  </td>\n",
-        "  <td>                                                                                               <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_clip.ipynb\">\n",
+        "  <td>\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/model_garden/model_garden_pytorch_clip.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "Open in Vertex AI Workbench\n",
        "    </a>\n",
+        "    (a Python-3 CPU notebook is recommended)\n",
        "  </td>\n",
        "</table>"
      ]
@@ -207,7 +207,7 @@
      "outputs": [],
      "source": [
        "# The pre-built serving docker image. It contains serving scripts and models.\n",
-        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve:latest\""
+        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve\""
      ]
    },
    {
@@ -282,7 +282,7 @@
        "        display_name=model_name,\n",
        "        serving_container_image_uri=SERVE_DOCKER_URI,\n",
        "        serving_container_ports=[7080],\n",
-        "        serving_container_predict_route=\"/predictions/diffusers_serving\",\n",
+        "        serving_container_predict_route=\"/predictions/transformers_serving\",\n",
        "        serving_container_health_route=\"/ping\",\n",
        "        serving_container_environment_variables=serving_env,\n",
        "        artifact_uri=artifact_uri,\n",
@@ -370,11 +370,11 @@
      },
      "outputs": [],
      "source": [
-        "# Delete models.\n",
-        "model.delete()\n",
-        "\n",
        "# Undeploy model and delete endpoint.\n",
-        "endpoint.delete(force=True)"
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
      ]
    }
  ],
@@ -37,18 +37,18 @@
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
-        "\n",
        "  <td>\n",
        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_controlnet.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
        "  </td>\n",
-        "  <td>                                                                                               <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_controlnet.ipynb\">\n",
+        "  <td>\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/model_garden/model_garden_pytorch_controlnet.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "Open in Vertex AI Workbench\n",
        "    </a>\n",
+        "    (a Python-3 CPU notebook is recommended)\n",
        "  </td>\n",
        "</table>"
      ]
@@ -257,7 +257,7 @@
        "TRAIN_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-diffusers-train:latest\"\n",
        "\n",
        "# The pre-built serving docker image. It contains serving scripts and models.\n",
-        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-diffusers-serve:latest\""
+        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-diffusers-serve\""
      ]
    },
    {
@@ -599,11 +599,11 @@
      },
      "outputs": [],
      "source": [
-        "# Delete models.\n",
-        "model.delete()\n",
-        "\n",
        "# Undeploy model and delete endpoint.\n",
-        "endpoint.delete(force=True)"
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
      ]
    }
  ],
@@ -37,18 +37,18 @@
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
-        "\n",
        "  <td>\n",
        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_instructpix2pix.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
        "  </td>\n",
-        "  <td>                                                                                               <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_instructpix2pix.ipynb\">\n",
+        "  <td>\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/model_garden/model_garden_pytorch_instructpix2pix.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "Open in Vertex AI Workbench\n",
        "    </a>\n",
+        "    (a Python-3 CPU notebook is recommended)\n",
        "  </td>\n",
        "</table>"
      ]
@@ -207,7 +207,7 @@
      "outputs": [],
      "source": [
        "# The pre-built serving docker image. It contains serving scripts and models.\n",
-        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-diffusers-serve:latest\""
+        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-diffusers-serve\""
      ]
    },
    {
@@ -392,11 +392,11 @@
      },
      "outputs": [],
      "source": [
-        "# Delete models.\n",
-        "model.delete()\n",
-        "\n",
        "# Undeploy model and delete endpoint.\n",
-        "endpoint.delete(force=True)"
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
      ]
    }
  ],
@@ -37,18 +37,18 @@
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
-        "\n",
        "  <td>\n",
        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_layoutml_document_qa.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
        "  </td>\n",
-        "  <td>                                                                                               <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_layoutml_document_qa.ipynb\">\n",
+        "  <td>\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/model_garden/model_garden_pytorch_layoutml_document_qa.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "Open in Vertex AI Workbench\n",
        "    </a>\n",
+        "    (a Python-3 CPU notebook is recommended)\n",
        "  </td>\n",
        "</table>"
      ]
@@ -207,7 +207,7 @@
      "outputs": [],
      "source": [
        "# The pre-built serving docker image. It contains serving scripts and models.\n",
-        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve:latest\""
+        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve\""
      ]
    },
    {
@@ -246,7 +246,7 @@
        "\n",
        "def download_image(url):\n",
        "    response = requests.get(url)\n",
-        "    return Image.open(BytesIO(response.content))\n",
+        "    return Image.open(BytesIO(response.content)).convert(\"RGB\")\n",
        "\n",
        "\n",
        "def image_to_base64(image, format=\"JPEG\"):\n",
@@ -282,7 +282,7 @@
        "        display_name=model_name,\n",
        "        serving_container_image_uri=SERVE_DOCKER_URI,\n",
        "        serving_container_ports=[7080],\n",
-        "        serving_container_predict_route=\"/predictions/diffusers_serving\",\n",
+        "        serving_container_predict_route=\"/predictions/transformers_serving\",\n",
        "        serving_container_health_route=\"/ping\",\n",
        "        serving_container_environment_variables=serving_env,\n",
        "        artifact_uri=artifact_uri,\n",
@@ -371,11 +371,11 @@
      },
      "outputs": [],
      "source": [
-        "# Delete models.\n",
-        "model.delete()\n",
-        "\n",
        "# Undeploy model and delete endpoint.\n",
-        "endpoint.delete(force=True)"
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
      ]
    }
  ],
@@ -37,18 +37,18 @@
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
-        "\n",
        "  <td>\n",
        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_owlvit.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
        "  </td>\n",
-        "  <td>                                                                                               <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_owlvit.ipynb\">\n",
+        "  <td>\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/model_garden/model_garden_pytorch_owlvit.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "Open in Vertex AI Workbench\n",
        "    </a>\n",
+        "    (a Python-3 CPU notebook is recommended)\n",
        "  </td>\n",
        "</table>"
      ]
@@ -207,7 +207,7 @@
      "outputs": [],
      "source": [
        "# The pre-built serving docker image. It contains serving scripts and models.\n",
-        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve:latest\""
+        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve\""
      ]
    },
    {
@@ -232,6 +232,8 @@
        "from datetime import datetime\n",
        "from io import BytesIO\n",
        "\n",
+        "import matplotlib.patches as patches\n",
+        "import matplotlib.pyplot as plt\n",
        "import requests\n",
        "from google.cloud import aiplatform\n",
        "from PIL import Image\n",
@@ -269,6 +271,23 @@
        "    return grid\n",
        "\n",
        "\n",
+        "def draw_image_with_boxes(image, boxes):\n",
+        "    fig, ax = plt.subplots()\n",
+        "    plt.axis(\"off\")\n",
+        "    ax.imshow(image)\n",
+        "    if len(boxes) == 0:\n",
+        "        return\n",
+        "    boxes = boxes[\"boxes\"]\n",
+        "    for box in boxes:\n",
+        "        x, y = box[\"xmin\"], box[\"ymin\"]\n",
+        "        width, height = box[\"xmax\"] - x, box[\"ymax\"] - y\n",
+        "        rect = patches.Rectangle(\n",
+        "            (x, y), width, height, linewidth=2, edgecolor=\"yellow\", facecolor=\"none\"\n",
+        "        )\n",
+        "        ax.add_patch(rect)\n",
+        "    plt.show()\n",
+        "\n",
+        "\n",
        "def deploy_model(model_id, task):\n",
        "    model_name = \"owl-vit\"\n",
        "    endpoint = aiplatform.Endpoint.create(display_name=f\"{model_name}-endpoint\")\n",
@@ -282,7 +301,7 @@
        "        display_name=model_name,\n",
        "        serving_container_image_uri=SERVE_DOCKER_URI,\n",
        "        serving_container_ports=[7080],\n",
-        "        serving_container_predict_route=\"/predictions/diffusers_serving\",\n",
+        "        serving_container_predict_route=\"/predictions/transformers_serving\",\n",
        "        serving_container_health_route=\"/ping\",\n",
        "        serving_container_environment_variables=serving_env,\n",
        "        artifact_uri=artifact_uri,\n",
@@ -346,7 +365,7 @@
        "    {\"image\": image_to_base64(image), \"text\": \"cat\"},\n",
        "]\n",
        "preds = endpoint.predict(instances=instances).predictions\n",
-        "draw_image_with_boxes(image, preds[0][\"boxes\"])\n",
+        "draw_image_with_boxes(image, preds[0])\n",
        "print(preds)"
      ]
    },
@@ -367,11 +386,11 @@
      },
      "outputs": [],
      "source": [
-        "# Delete models.\n",
-        "model.delete()\n",
-        "\n",
        "# Undeploy model and delete endpoint.\n",
-        "endpoint.delete(force=True)"
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
      ]
    }
  ],
@@ -37,18 +37,18 @@
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
-        "\n",
        "  <td>\n",
        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
        "  </td>\n",
-        "  <td>                                                                                               <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion.ipynb\">\n",
+        "  <td>\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "Open in Vertex AI Workbench\n",
        "    </a>\n",
+        "    (a Python-3 CPU notebook is recommended)\n",
        "  </td>\n",
        "</table>"
      ]
@@ -257,7 +257,7 @@
        "TRAIN_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-diffusers-train:latest\"\n",
        "\n",
        "# The pre-built serving docker image. It contains serving scripts and models.\n",
-        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-diffusers-serve:latest\""
+        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-diffusers-serve\""
      ]
    },
    {
@@ -546,11 +546,11 @@
      },
      "outputs": [],
      "source": [
-        "# Delete models.\n",
-        "model.delete()\n",
-        "\n",
        "# Undeploy model and delete endpoint.\n",
-        "endpoint.delete(force=True)"
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
      ]
    },
    {
@@ -625,11 +625,11 @@
      },
      "outputs": [],
      "source": [
-        "# Delete models.\n",
-        "model.delete()\n",
-        "\n",
        "# Undeploy model and delete endpoint.\n",
-        "endpoint.delete(force=True)"
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
      ]
    }
  ],
@@ -37,18 +37,18 @@
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
-        "\n",
        "  <td>\n",
        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_inpainting.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
        "  </td>\n",
-        "  <td>                                                                                               <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_inpainting.ipynb\">\n",
+        "  <td>\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/model_garden/model_garden_pytorch_stable_diffusion_inpainting.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "Open in Vertex AI Workbench\n",
        "    </a>\n",
+        "    (a Python-3 CPU notebook is recommended)\n",
        "  </td>\n",
        "</table>"
      ]
@@ -257,7 +257,7 @@
        "TRAIN_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-diffusers-train:latest\"\n",
        "\n",
        "# The pre-built serving docker image. It contains serving scripts and models.\n",
-        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-diffusers-serve:latest\""
+        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-diffusers-serve\""
      ]
    },
    {
@@ -554,11 +554,11 @@
      },
      "outputs": [],
      "source": [
-        "# Delete models.\n",
-        "model.delete()\n",
-        "\n",
        "# Undeploy model and delete endpoint.\n",
-        "endpoint.delete(force=True)"
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
      ]
    }
  ],
@@ -37,18 +37,18 @@
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
-        "\n",
        "  <td>\n",
        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_vilt_vqa.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
        "      View on GitHub\n",
        "    </a>\n",
        "  </td>\n",
-        "  <td>                                                                                               <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_vilt_vqa.ipynb\">\n",
+        "  <td>\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/model_garden/model_garden_pytorch_vilt_vqa.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "Open in Vertex AI Workbench\n",
        "    </a>\n",
+        "    (a Python-3 CPU notebook is recommended)\n",
        "  </td>\n",
        "</table>"
      ]
@@ -207,7 +207,7 @@
      "outputs": [],
      "source": [
        "# The pre-built serving docker image. It contains serving scripts and models.\n",
-        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve:latest\""
+        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve\""
      ]
    },
    {
@@ -282,7 +282,7 @@
        "        display_name=model_name,\n",
        "        serving_container_image_uri=SERVE_DOCKER_URI,\n",
        "        serving_container_ports=[7080],\n",
-        "        serving_container_predict_route=\"/predictions/diffusers_serving\",\n",
+        "        serving_container_predict_route=\"/predictions/transformers_serving\",\n",
        "        serving_container_health_route=\"/ping\",\n",
        "        serving_container_environment_variables=serving_env,\n",
        "        artifact_uri=artifact_uri,\n",
@@ -369,11 +369,11 @@
      },
      "outputs": [],
      "source": [
-        "# Delete models.\n",
-        "model.delete()\n",
-        "\n",
        "# Undeploy model and delete endpoint.\n",
-        "endpoint.delete(force=True)"
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
      ]
    }
  ],
@@ -37,7 +37,6 @@
        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
        "    </a>\n",
        "  </td>\n",
-        "\n",
        "  <td>\n",
        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_vit_gpt2_image_captioning.ipynb\">\n",
        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
@@ -45,10 +44,11 @@
        "    </a>\n",
        "  </td>\n",
        "  <td>                                                                                               <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/community/model_garden/model_garden_pytorch_vit_gpt2_image_captioning.ipynb\">\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/notebooks/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/community/model_garden/model_garden_pytorch_vit_gpt2_image_captioning.ipynb\">\n",
        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
        "Open in Vertex AI Workbench\n",
        "    </a>\n",
+        "    (a Python-3 CPU notebook is recommended)\n",
        "  </td>\n",
        "</table>"
      ]
@@ -207,7 +207,7 @@
      "outputs": [],
      "source": [
        "# The pre-built serving docker image. It contains serving scripts and models.\n",
-        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve:latest\""
+        "SERVE_DOCKER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/pytorch-transformers-serve\""
      ]
    },
    {
@@ -282,7 +282,7 @@
        "        display_name=model_name,\n",
        "        serving_container_image_uri=SERVE_DOCKER_URI,\n",
        "        serving_container_ports=[7080],\n",
-        "        serving_container_predict_route=\"/predictions/diffusers_serving\",\n",
+        "        serving_container_predict_route=\"/predictions/transformers_serving\",\n",
        "        serving_container_health_route=\"/ping\",\n",
        "        serving_container_environment_variables=serving_env,\n",
        "        artifact_uri=artifact_uri,\n",
@@ -367,11 +367,11 @@
      },
      "outputs": [],
      "source": [
-        "# Delete models.\n",
-        "model.delete()\n",
-        "\n",
        "# Undeploy model and delete endpoint.\n",
-        "endpoint.delete(force=True)"
+        "endpoint.delete(force=True)\n",
+        "\n",
+        "# Delete models.\n",
+        "model.delete()"
      ]
    }
  ],
@@ -405,7 +405,7 @@
        "                checkpoint_path = os.path.relpath(checkpoint_path, checkpoint_name)\n",
        "                break\n",
        "\n",
-        "    ! gsutil cp -r $checkpoint_name $CHECKPOINT_BUCKET\n",
+        "    ! gsutil cp -r $checkpoint_name $CHECKPOINT_BUCKET/\n",
        "    checkpoint_uri = os.path.join(CHECKPOINT_BUCKET, checkpoint_name, checkpoint_path)\n",
        "    print(\"Checkpoint uploaded to\", checkpoint_uri)\n",
        "    return checkpoint_uri"
@@ -501,7 +501,7 @@
        "* `input_file_type`: The input file type, such as csv or jsonl.\n",
        "* `split_ratio`: The proportion of data to split into train/validation/test.\n",
        "* `num_shard`: The number of shards for train/validation/test.\n",
-        "* `output_dir`: The output directory, which will container prepared train/test/validation data."
+        "* `output_dir`: The output directory, which will contain prepared train/test/validation data."
      ]
    },
    {
@@ -519,7 +519,7 @@
        ")\n",
        "\n",
        "input_file_path = \"\"  # @param {type:\"string\"}\n",
-        "input_file_type = \"\"  # @param {type:\"string\"}\n",
+        "input_file_type = \"csv\"  # @param ['csv', 'jsonl', 'coco_json']\n",
        "split_ratio = \"0.8,0.1,0.1\"\n",
        "num_shard = \"10,10,10\"\n",
        "data_converter_output_dir = os.path.join(BUCKET_URI, data_converter_job_name)\n",
@@ -582,7 +582,7 @@
      "source": [
        "#### Define the following specifications\n",
        "\n",
-        "* `worker_pool_specs`: Dictionary specifying the machine type and Docker image. This example defines a single node cluster with one `n1-standard-4` machine with two `NVIDIA_TESLA_T4` GPUs.\n",
+        "* `worker_pool_specs`: Dictionary specifying the machine type and Docker image. This example defines a single node cluster with one `n1-highmem-16` machine with two `NVIDIA_TESLA_V100` GPUs.\n",
        "* `parameter_spec`: Dictionary specifying the parameters to optimize. The dictionary key is the string assigned to the command line argument for each hyperparameter in your training application code, and the dictionary value is the parameter specification. The parameter specification includes the type, min/max values, and scale for the hyperparameter.\n",
        "* `metric_spec`: Dictionary specifying the metric to optimize. The dictionary key is the `hyperparameter_metric_tag` that you set in your training application code, and the value is the optimization goal."
      ]
@@ -651,7 +651,7 @@
        "        common_args,\n",
        "        **{\n",
        "            \"experiment\": \"retinanet_spinenet_coco\",\n",
-        "            \"config_file\": TRAIN_SPINENET96_CONFIG,\n",
+        "            \"config_file\": TRAIN_SPINENET143_CONFIG,\n",
        "            \"anchor_size\": 4,\n",
        "        },\n",
        "    ),\n",
@@ -162,13 +162,7 @@
      "outputs": [],
      "source": [
        "# Install the packages\n",
-        "import os\n",
-        "\n",
-        "if not os.getenv(\"IS_TESTING\"):\n",
-        "    USER = \"--user\"\n",
-        "else:\n",
-        "    USER = \"\"\n",
-        "! pip3 install {USER} --upgrade google-cloud-aiplatform"
+        "! pip3 install --upgrade --quiet google-cloud-aiplatform"
      ]
    },
    {
@@ -359,7 +353,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = \"gs://your-bucket-name-unique\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -40,4 +40,5 @@
 /model_evaluation/custom_tabular_classification_model_evaluation.ipynb @soheilazangeneh
 /sdk/SDK_FBProphet_Forecasting_Online.ipynb @brianchunkang
 /pipelines/Train_tabular_models_with_many_frameworks_and_import_to_Vertex_AI_using_Pipelines @Ark-kun
-/experiments/get_started_with_vertex_experiments_autologging.ipynb @inardini
+/experiments/get_started_with_vertex_experiments_autologging.ipynb @inardini
+/experiments/delete_outdated_tensorboard_experiments.ipynb @inardini
@@ -325,7 +325,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -319,7 +319,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -327,7 +327,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -329,7 +329,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -242,22 +242,7 @@
      "source": [
        "## Notes about service account and permission\n",
        "\n",
-        "**By default no configuration is required**, if you run into any permission related issue, please make sure the service accounts above have the required roles:\n",
-        "\n",
-        "|Service account email|Description|Roles|\n",
-        "|---|---|---|\n",
-        "|PROJECT_NUMBER-compute@developer.gserviceaccount.com|Compute Engine default service account|Dataflow Developer, Dataflow Worker, Storage Admin, BigQuery Data Editor, Vertex AI User, Service Account User|\n",
-        "|service-PROJECT_NUMBER@gcp-sa-aiplatform.iam.gserviceaccount.com|AI Platform Service Agent|Vertex AI Service Agent|\n",
-        "\n",
-        "\n",
-        "1. Goto https://console.cloud.google.com/iam-admin/iam.\n",
-        "2. Check the \"Include Google-provided role grants\" checkbox.\n",
-        "3. Find the above emails.\n",
-        "4. Grant the corresponding roles.\n",
-        "\n",
-        "### Using data source from a different project\n",
-        "- For the BQ data source, grant both service accounts the \"BigQuery Data Viewer\" role.\n",
-        "- For the CSV data source, grant both service accounts the \"Storage Object Viewer\" role.\n"
+        "**By default no configuration is required**, if you run into any permission related issue, please make sure the service accounts have the required roles listed in the [Service accounts for Tabular Workflow for End-to-End AutoML documentation](https://cloud.google.com/vertex-ai/docs/tabular-data/tabular-workflows/service-accounts#e2e-automl)."
      ]
    },
    {
@@ -301,7 +301,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -799,7 +799,7 @@
        "dataset.delete()\n",
        "\n",
        "# Delete training job\n",
-        "job.delete()"
+        "dag.delete()"
      ]
    }
  ],
@@ -298,7 +298,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -805,7 +805,7 @@
        "dataset.delete()\n",
        "\n",
        "# Delete training job\n",
-        "job.delete()"
+        "dag.delete()"
      ]
    }
  ],
@@ -319,7 +319,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -297,7 +297,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -0,0 +1,669 @@
+{
+  "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "ur8xi4C7S06n"
+      },
+      "outputs": [],
+      "source": [
+        "# Copyright 2022 Google LLC\n",
+        "#\n",
+        "# Licensed under the Apache License, Version 2.0 (the \"License\");\n",
+        "# you may not use this file except in compliance with the License.\n",
+        "# You may obtain a copy of the License at\n",
+        "#\n",
+        "#     https://www.apache.org/licenses/LICENSE-2.0\n",
+        "#\n",
+        "# Unless required by applicable law or agreed to in writing, software\n",
+        "# distributed under the License is distributed on an \"AS IS\" BASIS,\n",
+        "# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n",
+        "# See the License for the specific language governing permissions and\n",
+        "# limitations under the License."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "JAPoU8Sm5E6e"
+      },
+      "source": [
+        "# Delete Outdated Experiments in Vertex AI TensorBoard\n",
+        "\n",
+        "<table align=\"left\">\n",
+        "\n",
+        "  <td>\n",
+        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/experiments/delete_outdated_tensorboard_experiments.ipynb\">\n",
+        "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
+        "    </a>\n",
+        "  </td>\n",
+        "  <td>\n",
+        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/experiments/delete_outdated_tensorboard_experiments.ipynb\">\n",
+        "      <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
+        "      View on GitHub\n",
+        "    </a>\n",
+        "  </td>\n",
+        "  <td>\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/vertex-ai-samples/main/notebooks/official/experiments/delete_outdated_tensorboard_experiments.ipynb\">\n",
+        "      <img src=\"https://lh3.googleusercontent.com/UiNooY4LUgW_oTvpsNhPpQzsstV5W8F7rYgxgGBD85cWJoLmrOzhVs_ksK_vgx40SHs7jCqkTkCk=e14-rj-sc0xffffff-h130-w32\" alt=\"Vertex AI logo\">\n",
+        "      Open in Vertex AI Workbench\n",
+        "    </a>\n",
+        "  </td>\n",
+        "</table>"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "24743cf4a1e1"
+      },
+      "source": [
+        "**_NOTE_**: This notebook has been tested in the following environment:\n",
+        "\n",
+        "* Python version = 3.9"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "tvgnzT1CKxrO"
+      },
+      "source": [
+        "## Overview\n",
+        "\n",
+        "Vertex AI will have a new Vertex AI TensorBoard billing model. From August 2023, it will change from monthly `$300/user` to monthly `$10/GB`. In preparation for this change, users need to delete old Vertex AI TensorBoard Experiments to avoid unnecessary storage costs when the pricing change takes place.\n",
+        "\n",
+        "Learn more about [Vertex AI TensorBoard](https://cloud.google.com/vertex-ai/docs/experiments/tensorboard-overview)."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "d975e698c9a4"
+      },
+      "source": [
+        "### Objective\n",
+        "\n",
+        "In this tutorial, you learn how to delete outdated TensorBoard Experiments to avoid unnecessary storage costs.\n",
+        "\n",
+        "This tutorial uses the following Google Cloud ML services and resources:\n",
+        "\n",
+        "- Vertex AI Tensorboard\n",
+        "\n",
+        "The steps performed include:\n",
+        "\n",
+        "- How to delete the TB Experiment with a predefined key-value label pair `<label_key, label_value>`\n",
+        "\n",
+        "- How to delete the TB Experiments created before the  `create_time`\n",
+        "\n",
+        "- How to delete the TB Experiments created before the  `update_time`"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "08d289fa873f"
+      },
+      "source": [
+        "### Dataset\n",
+        "\n",
+        "No dataset is used."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "aed92deeb4a0"
+      },
+      "source": [
+        "### Costs\n",
+        "\n",
+        "This tutorial uses billable components of Google Cloud:\n",
+        "\n",
+        "* Vertex AI\n",
+        "* Cloud Storage\n",
+        "\n",
+        "Learn about [Vertex AI pricing](https://cloud.google.com/vertex-ai/pricing), and [Cloud Storage pricing](https://cloud.google.com/storage/pricing),\n",
+        "and use the [Pricing Calculator](https://cloud.google.com/products/calculator/)\n",
+        "to generate a cost estimate based on your projected usage."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "i7EUnXsZhAGF"
+      },
+      "source": [
+        "## Installation\n",
+        "\n",
+        "Install the following packages required to execute this notebook."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "2b4ef9b72d43"
+      },
+      "outputs": [],
+      "source": [
+        "# Install the packages\n",
+        "import os\n",
+        "\n",
+        "if not os.getenv(\"IS_TESTING\"):\n",
+        "    USER = \"--user\"\n",
+        "else:\n",
+        "    USER = \"\"\n",
+        "! pip3 install {USER} --upgrade google-cloud-aiplatform"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "58707a750154"
+      },
+      "source": [
+        "### Colab only: Uncomment the following cell to restart the kernel."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "f200f10a1da3"
+      },
+      "outputs": [],
+      "source": [
+        "# Automatically restart kernel after installs so that your environment can access the new packages\n",
+        "# import IPython\n",
+        "\n",
+        "# app = IPython.Application.instance()\n",
+        "# app.kernel.do_shutdown(True)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "BF1j6f9HApxa"
+      },
+      "source": [
+        "## Before you begin\n",
+        "\n",
+        "### Set up your Google Cloud project\n",
+        "\n",
+        "**The following steps are required, regardless of your notebook environment.**\n",
+        "\n",
+        "1. [Select or create a Google Cloud project](https://console.cloud.google.com/cloud-resource-manager). When you first create an account, you get a $300 free credit towards your compute/storage costs.\n",
+        "\n",
+        "2. [Make sure that billing is enabled for your project](https://cloud.google.com/billing/docs/how-to/modify-project).\n",
+        "\n",
+        "3. [Enable the Vertex AI API](https://console.cloud.google.com/flows/enableapi?apiid=aiplatform.googleapis.com).\n",
+        "\n",
+        "4. If you are running this notebook locally, you need to install the [Cloud SDK](https://cloud.google.com/sdk)."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "WReHDGG5g0XY"
+      },
+      "source": [
+        "#### Set your project ID\n",
+        "\n",
+        "**If you don't know your project ID**, try the following:\n",
+        "* Run `gcloud config list`.\n",
+        "* Run `gcloud projects list`.\n",
+        "* See the support page: [Locate the project ID](https://support.google.com/googleapi/answer/7014113)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "oM1iC_MfAts1"
+      },
+      "outputs": [],
+      "source": [
+        "PROJECT_ID = \"[your-project-id]\"  # @param {type:\"string\"}\n",
+        "\n",
+        "# Set the project id\n",
+        "! gcloud config set project {PROJECT_ID}"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "region"
+      },
+      "source": [
+        "#### Region\n",
+        "\n",
+        "You can also change the `REGION` variable used by Vertex AI. Learn more about [Vertex AI regions](https://cloud.google.com/vertex-ai/docs/general/locations)."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "QAup21nz6LHk"
+      },
+      "outputs": [],
+      "source": [
+        "REGION = \"us-central1\"  # @param {type: \"string\"}"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "sBCra4QMA2wR"
+      },
+      "source": [
+        "### Authenticate your Google Cloud account\n",
+        "\n",
+        "Depending on your Jupyter environment, you may have to manually authenticate. Follow the relevant instructions below."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "74ccc9e52986"
+      },
+      "source": [
+        "**1. Vertex AI Workbench**\n",
+        "* Do nothing as you are already authenticated."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "de775a3773ba"
+      },
+      "source": [
+        "**2. Local JupyterLab instance, uncomment and run:**"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "254614fa0c46"
+      },
+      "outputs": [],
+      "source": [
+        "# ! gcloud auth login"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "ef21552ccea8"
+      },
+      "source": [
+        "**3. Colab, uncomment and run:**"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "603adbbf0532"
+      },
+      "outputs": [],
+      "source": [
+        "# from google.colab import auth\n",
+        "# auth.authenticate_user()"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "f6b2ccc891ed"
+      },
+      "source": [
+        "**4. Service account or other**\n",
+        "* See how to grant permissions to your service account at https://cloud.google.com/marketplace/docs/grant-service-account-access."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "960505627ddf"
+      },
+      "source": [
+        "### Import libraries"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "PyQmSRbKA8r-"
+      },
+      "outputs": [],
+      "source": [
+        "from google.cloud import aiplatform"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "NUJOpq7_-nt6"
+      },
+      "source": [
+        "### Define constants\n",
+        "\n",
+        "Define variables you use in this tutorial. In particular, you set\n",
+        "\n",
+        "- `CREATE_TIME_CUT` : delete tensorboard experiments that were created before CREATE_TIME_CUT. For example, `2022-12-31`.\n",
+        "\n",
+        "- `UPDATE_TIME_CUT` : delete tensorboard experiments that were created before UPDATE_TIME_CUT. For example, `2022-12-31`.\n",
+        "\n",
+        "- `DETAILED_LOG` : a booled variable to see Tensorboard deletion progress. If True, it shows progress by experiments. Otherwise, it reports progress per 100 experiments."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "0AkyoMre-qe5"
+      },
+      "outputs": [],
+      "source": [
+        "CREATE_TIME_CUT = \"\"  # @param {type:\"string\"}\n",
+        "\n",
+        "UPDATE_TIME_CUT = \"\"  # @param {type:\"string\"}\n",
+        "\n",
+        "DETAILED_LOG = True  # @param {type: \"boolean\"}"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "n68J5KJ4ERbF"
+      },
+      "source": [
+        "### Define Helpers"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "JIXC-tVrETET"
+      },
+      "outputs": [],
+      "source": [
+        "def clean_up_by_label(tensorboard_instance, label_key, label_value):\n",
+        "    \"\"\"Delete the TB Experiment with the key-value label pair <label_key, label_value>\"\"\"\n",
+        "    # List tensorboard experiments\n",
+        "    tensorboard_experiments = aiplatform.TensorboardExperiment.list(\n",
+        "        tensorboard_name=tensorboard_instance.resource_name\n",
+        "    )\n",
+        "    # Get the number of tensorboard experiments\n",
+        "    num_tensorboard_experiments = len(tensorboard_experiments)\n",
+        "    # For each experiment\n",
+        "    for i in range(num_tensorboard_experiments):\n",
+        "        tensorboard_experiment = tensorboard_experiments[i]\n",
+        "        if detailed_log or (i % 100 == 0):\n",
+        "            print(\n",
+        "                f\">>>checking TB experiment [{i + 1}/{num_tensorboard_experiments}]: {tensorboard_experiment.resource_name}\"\n",
+        "            )\n",
+        "        # Get experiment labels\n",
+        "        labels = tensorboard_experiment.labels\n",
+        "        # Filter by label\n",
+        "        if label_key in labels and labels[label_key] == label_value:\n",
+        "            # Delete experiment\n",
+        "            tensorboard_experiment.delete()\n",
+        "\n",
+        "\n",
+        "def clean_up_by_create_time(tensorboard_instance, create_time_cut):\n",
+        "    \"\"\"Delete the TB Experiment with the `create_time`\"\"\"\n",
+        "    # List tensorboard experiments\n",
+        "    tensorboard_experiments = aiplatform.TensorboardExperiment.list(\n",
+        "        tensorboard_name=tensorboard_instance.resource_name, order_by=\"create_time\"\n",
+        "    )\n",
+        "    # Get the number of tensorboard experiments\n",
+        "    num_tensorboard_experiments = len(tensorboard_experiments)\n",
+        "    # For each experiment\n",
+        "    for i in range(num_tensorboard_experiments):\n",
+        "        tensorboard_experiment = tensorboard_experiments[i]\n",
+        "        if detailed_log or (i % 100 == 0):\n",
+        "            print(\n",
+        "                f\">>> checking TB experiment [{i + 1}/{num_tensorboard_experiments}]: {tensorboard_experiment.resource_name}\"\n",
+        "            )\n",
+        "        # Filter by create_time\n",
+        "        if str(tensorboard_experiment.create_time) < create_time_cut:\n",
+        "            # Delete experiment\n",
+        "            tensorboard_experiment.delete()\n",
+        "        else:\n",
+        "            break\n",
+        "\n",
+        "\n",
+        "def clean_up_by_update_time(tensorboard_instance, update_time_cut):\n",
+        "    \"\"\"Delete the TB Experiment with with the `update_time`\"\"\"\n",
+        "    # List tensorboard experiments\n",
+        "    tensorboard_experiments = aiplatform.TensorboardExperiment.list(\n",
+        "        tensorboard_name=tensorboard_instance.resource_name, order_by=\"update_time\"\n",
+        "    )\n",
+        "    # Get the number of tensorboard experiments\n",
+        "    num_tensorboard_experiments = len(tensorboard_experiments)\n",
+        "    # For each experiment\n",
+        "    for i in range(num_tensorboard_experiments):\n",
+        "        tensorboard_experiment = tensorboard_experiments[i]\n",
+        "        if detailed_log or (i % 100 == 0):\n",
+        "            print(\n",
+        "                f\">>> checking TB experiment [{i + 1}/{num_tensorboard_experiments}]: {tensorboard_experiment.resource_name}\"\n",
+        "            )\n",
+        "        # Filter by update_time\n",
+        "        if str(tensorboard_experiment.update_time) < update_time_cut:\n",
+        "            tensorboard_experiment.delete()\n",
+        "        else:\n",
+        "            break"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "UCUOlTLYCT0B"
+      },
+      "source": [
+        "## Delete outdated Vertex AI TensorBoard Experiments"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "EGfFVHhWDjkl"
+      },
+      "source": [
+        "### Initialize Vertex AI SDK for Python\n",
+        "\n",
+        "Initialize the Vertex AI SDK for Python for your project."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "G1l-7Wft3jb6"
+      },
+      "outputs": [],
+      "source": [
+        "aiplatform.init(project=PROJECT_ID, location=REGION)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "e4672ef33691"
+      },
+      "source": [
+        "### Set delete outdated TensorBoard experiments \n",
+        "\n",
+        "Initialize a flag variable to start deleting outdated TensorBoard experiments and a flag variable to choose the deleting method. "
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "8cdc326c9823"
+      },
+      "outputs": [],
+      "source": [
+        "delete_tb_experiments = False  # @param {type: \"boolean\"}\n",
+        "\n",
+        "delete_method = \"\"  # @param [\"by_label\", \"by_create_time\", \"by_update_time\"]"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "JjJVARe5V-MG"
+      },
+      "source": [
+        "### Delete an TensorBoard instance\n",
+        "\n",
+        "To delete a Tensorboard instance, you need `TENSORBOARD_INSTANCE` ID which uniquely identifies the Tensorboard instance where you run experiments.\n",
+        "\n",
+        "To get the Tensorboard instance ID, you can either\n",
+        "\n",
+        "- go to the cloud console UI, Vertex AI > Experiments > Tensorboard Instances, or\n",
+        "- use the list command below to list all TensorBoard instances for your project and region.\n"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "eqM2pIO8_4Y7"
+      },
+      "outputs": [],
+      "source": [
+        "tensorboard_instances = aiplatform.Tensorboard.list(project=PROJECT_ID, location=REGION)\n",
+        "print(tensorboard_instances)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "wT_uPArACMPX"
+      },
+      "source": [
+        "Set the tensorboard instance id for which you want to delete experiments."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "hUync8E9CLcm"
+      },
+      "outputs": [],
+      "source": [
+        "if delete_tb_experiments:\n",
+        "\n",
+        "    TENSORBOARD_INSTANCE_ID = \"\"  # @param {type:\"string\"}\n",
+        "\n",
+        "    TENSORBOARD_INSTANCE = aiplatform.Tensorboard(\n",
+        "        project=PROJECT_ID, location=REGION, tensorboard_name=TENSORBOARD_INSTANCE_ID\n",
+        "    )\n",
+        "    print(TENSORBOARD_INSTANCE)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "OOp22fxdHqmT"
+      },
+      "source": [
+        "### Delete a TensorBoard Experiment with the key-value label pair\n",
+        "\n",
+        "You delete a TensorBoard experiment using a predefined `label_key` and `label_value`. For example, you may have assigned `delete` label key and `true` label value to indicate all Tensorboard experiments you want to delete.\n"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "N7KajCsjij1g"
+      },
+      "outputs": [],
+      "source": [
+        "LABEL_KEY = \"delete\"  # @param {type:\"string\"}\n",
+        "LABEL_VALUE = \"true\"  # @param {type:\"string\"}"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "N45samDzWBO2"
+      },
+      "outputs": [],
+      "source": [
+        "if delete_tb_experiments and delete_method == \"by_label\":\n",
+        "    clean_up_by_label(TENSORBOARD_INSTANCE, LABEL_KEY, LABEL_VALUE)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "XrEVqlgSIL6K"
+      },
+      "source": [
+        "### Delete a TensorBoard Experiment with `create_time`\n",
+        "\n",
+        "You delete a TensorBoard experiment using `create_time` field"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "4_bjNeQ7p87A"
+      },
+      "outputs": [],
+      "source": [
+        "if delete_tb_experiments and delete_method == \"by_create_time\":\n",
+        "    clean_up_by_create_time(TENSORBOARD_INSTANCE, CREATE_TIME_CUT)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "ZGNlSbIrIP_o"
+      },
+      "source": [
+        "### Delete a TensorBoard Experiment with `update_time`\n",
+        "\n",
+        "You delete a TensorBoard Experiment using a predefined `update_time` field"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "SDyc_a8XwEve"
+      },
+      "outputs": [],
+      "source": [
+        "if delete_tb_experiments and delete_method == \"by_update_time\":\n",
+        "    clean_up_by_update_time(TENSORBOARD_INSTANCE, UPDATE_TIME_CUT)"
+      ]
+    }
+  ],
+  "metadata": {
+    "colab": {
+      "name": "delete_outdated_tensorboard_experiments.ipynb",
+      "toc_visible": true
+    },
+    "kernelspec": {
+      "display_name": "Python 3",
+      "name": "python3"
+    }
+  },
+  "nbformat": 4,
+  "nbformat_minor": 0
+}
@@ -1577,7 +1577,7 @@
  ],
  "metadata": {
    "colab": {
-      "name": "UJ10 Vertex SDK Custom Scikit-Learn with pre-built training container.ipynb",
+      "name": "sdk-custom-scikit-learn-prebuilt-container.ipynb",
      "toc_visible": true
    },
    "kernelspec": {
@@ -604,12 +604,18 @@
        "        epochs=params[\"epochs\"],\n",
        "        dropout_rate=params[\"dropout_rate\"],\n",
        "    )\n",
-        "    aiplatform.log_metrics(\n",
-        "        {metric: values[-1] for metric, values in history.history.items()}\n",
-        "    )\n",
+        "\n",
+        "    for metric, values in history.history.items():\n",
+        "        try:\n",
+        "            aiplatform.log_metrics({metric: values[-1]})\n",
+        "        except:\n",
+        "            aiplatform.log_metrics({metric: 0.0})\n",
        "\n",
        "    loss, mae, mse = model.evaluate(normed_test_data, test_labels, verbose=2)\n",
-        "    aiplatform.log_metrics({\"eval_loss\": loss, \"eval_mae\": mae, \"eval_mse\": mse})"
+        "    try:\n",
+        "        aiplatform.log_metrics({\"eval_loss\": loss, \"eval_mae\": mae, \"eval_mse\": mse})\n",
+        "    except:\n",
+        "        aiplatform.log_metrics({\"eval_loss\": 0.0, \"eval_mae\": 0.0, \"eval_mse\": 0.0})"
      ]
    },
    {
@@ -330,7 +330,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -201,8 +201,8 @@
        "\n",
        "! pip3 install --upgrade {USER_FLAG} google-cloud-aiplatform \\\n",
        "                                    google-cloud-storage \\\n",
-        "                                    kfp \\\n",
-        "                                    google-cloud-pipeline-components -q"
+        "                                    'kfp<2' \\\n",
+        "                                    'google-cloud-pipeline-components<2' -q"
      ]
    },
    {
@@ -165,11 +165,11 @@
        "# Install the packages\n",
        "USER=''\n",
        "! pip3 install {USER} --upgrade google-cloud-aiplatform \\\n",
-        "                                google-cloud-pipeline-components\n",
+        "                                'google-cloud-pipeline-components<2'\n",
        "! pip3 install {USER}           tensorflow==2.5 \\\n",
        "                                tensorflow_hub\n",
        "  \n",
-        "! pip3 install {USER} --upgrade kfp"
+        "! pip3 install {USER} --upgrade 'kfp<2'"
      ]
    },
    {
@@ -362,7 +362,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -175,7 +175,7 @@
        "\n",
        "! pip3 install --upgrade google-cloud-aiplatform {USER_FLAG} -q\n",
        "! pip3 install -U google-cloud-storage {USER_FLAG} -q\n",
-        "! pip3 install {USER_FLAG} kfp google-cloud-pipeline-components --upgrade -q"
+        "! pip3 install {USER_FLAG} 'kfp<2' 'google-cloud-pipeline-components<2' --upgrade -q"
      ]
    },
    {
@@ -197,10 +197,10 @@
        "\n",
        "! pip3 install --upgrade google-cloud-aiplatform {USER_FLAG} -q\n",
        "! pip3 install -U google-cloud-storage {USER_FLAG} -q\n",
-        "! pip3 install {USER_FLAG} kfp google-cloud-pipeline-components --upgrade -q\n",
+        "! pip3 install {USER_FLAG} 'kfp<2' 'google-cloud-pipeline-components<2' --upgrade -q\n",
        "\n",
        "\n",
-        "! pip3 install --upgrade --force-reinstall $USER_FLAG tensorflow kfp google-cloud-aiplatform google-cloud-storage google-cloud-pipeline-components -q"
+        "! pip3 install --upgrade --force-reinstall $USER_FLAG tensorflow 'kfp<2' google-cloud-aiplatform google-cloud-storage 'google-cloud-pipeline-components<2' -q"
      ]
    },
    {
@@ -224,8 +224,8 @@
        "                            google-cloud-bigquery \\\n",
        "                            pandas \\\n",
        "                            pyarrow \\\n",
-        "                            kfp \\\n",
-        "                            google-cloud-pipeline-components {USER_FLAG} -q \n",
+        "                            'kfp<2' \\\n",
+        "                            'google-cloud-pipeline-components<2' {USER_FLAG} -q \n",
        "! pip3 install db-dtypes {USER_FLAG} -q"
      ]
    },
@@ -149,8 +149,8 @@
        "! pip3 install -U tensorflow-transform==1.2  -q\n",
        "! pip3 install -U tensorflow-io==0.18  -q\n",
        "! pip3 install --upgrade google-cloud-aiplatform[tensorboard] -q\n",
-        "! pip3 install --upgrade google-cloud-pipeline-components  -q\n",
-        "! pip3 install --upgrade kfp -q"
+        "! pip3 install --upgrade 'google-cloud-pipeline-components<2'  -q\n",
+        "! pip3 install --upgrade 'kfp<2' -q"
      ]
    },
    {
@@ -307,7 +307,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -193,7 +193,7 @@
        "if IS_WORKBENCH_NOTEBOOK:\n",
        "    USER_FLAG = \"--user\"\n",
        "\n",
-        "! pip3 install --upgrade --quiet {USER_FLAG} google-cloud-aiplatform kfp google-cloud-pipeline-components google-cloud-storage"
+        "! pip3 install --upgrade --quiet {USER_FLAG} google-cloud-aiplatform 'kfp<2' 'google-cloud-pipeline-components<2' google-cloud-storage"
      ]
    },
    {
@@ -198,7 +198,7 @@
        "\n",
        "! pip3 install --upgrade google-cloud-aiplatform \\\n",
        "                        google-cloud-storage \\\n",
-        "                        $USER kfp google-cloud-pipeline-components -q "
+        "                        $USER 'kfp<2' 'google-cloud-pipeline-components<2' -q "
      ]
    },
    {
@@ -196,7 +196,7 @@
        "\n",
        "! pip3 install --upgrade google-cloud-aiplatform {USER_FLAG} -q\n",
        "! pip3 install -U google-cloud-storage $USER_FLAG -q\n",
-        "! pip3 install $USER kfp google-cloud-pipeline-components --upgrade -q"
+        "! pip3 install $USER 'kfp<2' 'google-cloud-pipeline-components<2' --upgrade -q"
      ]
    },
    {
@@ -198,8 +198,8 @@
        "\n",
        "! pip3 install --upgrade {USER_FLAG} -q google-cloud-aiplatform \\\n",
        "                                        google-cloud-storage {USER_FLAG} \\\n",
-        "                                        kfp \\\n",
-        "                                        google-cloud-pipeline-components"
+        "                                        'kfp<2' \\\n",
+        "                                        'google-cloud-pipeline-components<2'"
      ]
    },
    {
@@ -199,8 +199,8 @@
        "\n",
        "\n",
        "! pip3 install --upgrade google-cloud-aiplatform \\\n",
-        "                        google-cloud-pipeline-components \\\n",
-        "                        kfp $USER_FLAG -q"
+        "                        'google-cloud-pipeline-components<2' \\\n",
+        "                        'kfp<2' $USER_FLAG -q"
      ]
    },
    {
@@ -203,7 +203,7 @@
        "\n",
        "! pip3 install --upgrade google-cloud-aiplatform {USER_FLAG} -q\n",
        "! pip3 install -U google-cloud-storage {USER_FLAG} -q\n",
-        "! pip3 install {USER_FLAG} kfp google-cloud-pipeline-components --upgrade -q"
+        "! pip3 install {USER_FLAG} 'kfp<2' 'google-cloud-pipeline-components<2' --upgrade -q"
      ]
    },
    {
@@ -201,7 +201,7 @@
        "\n",
        "! pip3 install --upgrade google-cloud-aiplatform {USER_FLAG} -q\n",
        "! pip3 install -U google-cloud-storage {USER_FLAG} -q\n",
-        "! pip3 install {USER_FLAG} kfp google-cloud-pipeline-components --upgrade -q\n",
+        "! pip3 install {USER_FLAG} 'kfp<2' 'google-cloud-pipeline-components<2' --upgrade -q\n",
        "\n",
        "if os.getenv(\"IS_TESTING\"):\n",
        "    ! pip3 install --upgrade matplotlib $USER_FLAG -q"
@@ -168,11 +168,11 @@
        "# Install the packages\n",
        "USER=''\n",
        "! pip3 install {USER} --upgrade google-cloud-aiplatform \\\n",
-        "                                google-cloud-pipeline-components\n",
+        "                                'google-cloud-pipeline-components<2'\n",
        "! pip3 install {USER}           tensorflow==2.5 \\\n",
        "                                tensorflow_hub\n",
        "\n",
-        "! pip3 install {USER} --upgrade kfp"
+        "! pip3 install {USER} --upgrade 'kfp<2'"
      ]
    },
    {
@@ -363,7 +363,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -128,8 +128,8 @@
        "# Install the packages\n",
        "! pip3 install --upgrade google-cloud-aiplatform \\\n",
        "                        google-cloud-storage \\\n",
-        "                        kfp \\\n",
-        "                        google-cloud-pipeline-components"
+        "                        'kfp<2' \\\n",
+        "                        'google-cloud-pipeline-components<2'"
      ]
    },
    {
@@ -268,22 +268,7 @@
      "source": [
        "## Notes about service account and permission\n",
        "\n",
-        "**By default no configuration is required**, if you run into any permission related issue, please make sure the service accounts above have the required roles:\n",
-        "\n",
-        "|Service account email|Description|Roles|\n",
-        "|---|---|---|\n",
-        "|PROJECT_NUMBER-compute@developer.gserviceaccount.com|Compute Engine default service account|Dataflow Admin, Dataflow Worker, Storage Admin, BigQuery Admin, Vertex AI User|\n",
-        "|service-PROJECT_NUMBER@gcp-sa-aiplatform.iam.gserviceaccount.com|AI Platform Service Agent|Vertex AI Service Agent|\n",
-        "\n",
-        "\n",
-        "1. Goto https://console.cloud.google.com/iam-admin/iam.\n",
-        "2. Check the \"Include Google-provided role grants\" checkbox.\n",
-        "3. Find the above emails.\n",
-        "4. Grant the corresponding roles.\n",
-        "\n",
-        "### Using data source from a different project\n",
-        "- For the BQ data source, grant both service accounts the \"BigQuery Data Viewer\" role.\n",
-        "- For the CSV data source, grant both service accounts the \"Storage Object Viewer\" role.\n"
+        "**By default no configuration is required**, if you run into any permission related issue, please make sure the service accounts have the required roles listed in the [Service accounts for Tabular Workflow for TabNet, and Tabular Workflow for Wide & Deep, and Prophet documentation](https://cloud.google.com/vertex-ai/docs/tabular-data/tabular-workflows/service-accounts#fte-workflow)."
      ]
    },
    {
@@ -268,22 +268,7 @@
      "source": [
        "## Notes about service account and permission\n",
        "\n",
-        "**By default no configuration is required**, if you run into any permission related issue, please make sure the service accounts above have the required roles:\n",
-        "\n",
-        "|Service account email|Description|Roles|\n",
-        "|---|---|---|\n",
-        "|PROJECT_NUMBER-compute@developer.gserviceaccount.com|Compute Engine default service account|Dataflow Admin, Dataflow Worker, Storage Admin, BigQuery Admin, Vertex AI User|\n",
-        "|service-PROJECT_NUMBER@gcp-sa-aiplatform.iam.gserviceaccount.com|AI Platform Service Agent|Vertex AI Service Agent|\n",
-        "\n",
-        "\n",
-        "1. Goto https://console.cloud.google.com/iam-admin/iam.\n",
-        "2. Check the \"Include Google-provided role grants\" checkbox.\n",
-        "3. Find the above emails.\n",
-        "4. Grant the corresponding roles.\n",
-        "\n",
-        "### Using data source from a different project\n",
-        "- For the BQ data source, grant both service accounts the \"BigQuery Data Viewer\" role.\n",
-        "- For the CSV data source, grant both service accounts the \"Storage Object Viewer\" role."
+        "**By default no configuration is required**, if you run into any permission related issue, please make sure the service accounts have the required roles listed in the [Service accounts for Tabular Workflow for TabNet, and Tabular Workflow for Wide & Deep, and Prophet documentation](https://cloud.google.com/vertex-ai/docs/tabular-data/tabular-workflows/service-accounts#fte-workflow)."
      ]
    },
    {
@@ -296,7 +296,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
@@ -300,7 +300,7 @@
      },
      "outputs": [],
      "source": [
-        "BUCKET_URI = f\"gs://your-bucket-name-unique-{PROJECT_ID}\"  # @param {type:\"string\"}"
+        "BUCKET_URI = f\"gs://your-bucket-name-{PROJECT_ID}-unique\"  # @param {type:\"string\"}"
      ]
    },
    {
Author	SHA1	Message	Date
Andrew Ferlitsch	5c49bc6fe7	fix: rename UJ10	2023-04-19 23:51:55 +00:00
Connor McCarthyandGitHub	098ad42f70	pin kfp in google_cloud_pipeline_components_automl_text (#1748 )	2023-04-19 23:01:21 +00:00
Connor McCarthyandGitHub	1f141d0059	pin kfp in google_cloud_pipeline_components_automl_images (#1746 )	2023-04-19 19:29:50 +00:00
Andrew FerlitschandGitHub	e0a1608783	fix: pin kfp and gcpc < 2.0 (#1738 )	2023-04-19 18:37:16 +00:00
Connor McCarthyandGitHub	1a469a09df	pin kfp in google_cloud_pipeline_components_automl_tabular (#1747 )	2023-04-19 18:14:04 +00:00
Connor McCarthyandGitHub	bc3f6d3819	pin kfp in custom_model_training_and_batch_prediction (#1741 )	2023-04-19 16:53:23 +00:00
Andrew FerlitschandGitHub	f088a3ce69	update (#1736 )	2023-04-19 16:52:34 +00:00
Connor McCarthyandGitHub	863660042b	pin kfp in custom_tabular_train_batch_pred_bq_pipeline (#1742 )	2023-04-19 16:51:11 +00:00
Connor McCarthyandGitHub	7162248267	pin kfp in google_cloud_pipeline_components_model_train_upload_deploy (#1749 )	2023-04-19 16:50:43 +00:00
Connor McCarthyandGitHub	2e908a8efa	pin kfp in get_started_with_hpt_pipeline_components (#1743 )	2023-04-19 16:50:13 +00:00
Connor McCarthyandGitHub	588d2c880e	pin kfp in get_started_with_machine_management (#1745 )	2023-04-19 16:49:12 +00:00
Connor McCarthyandGitHub	54f10c7411	pin kfp in google_cloud_pipeline_components_model_upload_predict_evaluate (#1750 )	2023-04-19 16:48:46 +00:00
Connor McCarthyandGitHub	0c2b9e9f45	pin kfp in multicontender_vs_champion_deployment_method (#1753 )	2023-04-19 16:48:21 +00:00
Connor McCarthyandGitHub	fe07d416e5	pin kfp in challenger_vs_blessed_deployment_method (#1739 )	2023-04-19 16:04:05 +00:00
Connor McCarthyandGitHub	9c69cbfd54	pin kfp in control_flow_kfp (#1740 )	2023-04-19 16:03:33 +00:00
Connor McCarthyandGitHub	2a25be6af2	pin kfp in lightweight_functions_component_io_kfp (#1751 )	2023-04-19 16:02:52 +00:00
Connor McCarthyandGitHub	9680a7e772	pin kfp in metrics_viz_run_compare_kfp (#1752 )	2023-04-19 16:02:03 +00:00
Connor McCarthyandGitHub	30527875ed	pin kfp in pipelines_intro_kfp (#1754 )	2023-04-19 16:01:36 +00:00
yishan-pu GitHub Yishan Pu	8c11e19d86	Update service accounts and permissions info in the Wide and Deep notebook (#1737 ) Co-authored-by: Yishan Pu <yishanpu@google.com>	2023-04-19 15:24:34 +00:00
yishan-pu GitHub Yishan Pu	f44f51c06d	Update the service accounts and permissions info in the TabNet notebook (#1734 ) Co-authored-by: Yishan Pu <yishanpu@google.com>	2023-04-18 16:06:42 +00:00
yishan-pu GitHub Yishan Pu	840b537ea6	Update the E2E AutoML Notebook regarding the service accounts and permissions info (#1731 ) Co-authored-by: Yishan Pu <yishanpu@google.com>	2023-04-17 23:16:28 +00:00
Xiang XuandGitHub	d2602b944f	Add blip2 notebook (#1730 ) * add blip2 * resolve comments	2023-04-17 23:14:34 +00:00
Xiang XuandGitHub	bb39135946	fix owlvit (#1729 )	2023-04-17 15:19:36 +00:00
gericdongandGitHub	cb72a56e55	feat: PyTorch train and deploy E2E with pre-built containers (#1728 ) * feat: PyTorch train and deploy E2E with pre-built containers * address review comments * address review comment 2 * suppress gsutil warning messags * Use unique names * Corrected the project id template	2023-04-17 14:56:43 +00:00
Andrew FerlitschandGitHub	eee97362d0	fix: migrate tabnet (#1727 ) * fix: migrate tabnet * fix: describe * updates from review	2023-04-13 21:09:23 +00:00
Ivan NardiniandGitHub	a92c5be0e9	feat: delete outdated tensorboard experiments (#1723 ) * add delete outdated tensorboard experiments notebook * update CODEOWNERS * fix aiplatform import * linter passed * set a flag varible to pass test * linter passed * add andy review * linter passed	2023-04-13 15:57:43 +00:00
Xiang XuandGitHub	80707a3ba7	fix clean (#1725 )	2023-04-12 20:28:00 +00:00
Eric Schmidt GitHub Andrew Ferlitsch	903f69b81c	feat: PyTorch training with GCS data (#1705 ) * feat: PyTorch training with GCS data * fix kernel restart * per reviewer * Added requirements section * fixes for build * lint, build * per reviewer * linter again * per reviewer * per reviewer, linter --------- Co-authored-by: Andrew Ferlitsch <aferlitsch@google.com>	2023-04-12 17:39:39 +00:00
Andrew FerlitschandGitHub	40a31daaef	Revert "update: thread safe limiter (#1718 )" (#1724 ) This reverts commit `37ba322412`.	2023-04-12 16:33:16 +00:00
Andrew FerlitschandGitHub	37ba322412	update: thread safe limiter (#1718 ) * update: thread safe limiter * update: thread safe limiter * fix: missing install	2023-04-12 00:05:36 +00:00
Xiang XuandGitHub	ee43400e89	fix clean (#1719 )	2023-04-11 22:59:00 +00:00
Xiang XuandGitHub	176c3721fd	fix pytorch notebooks (#1717 )	2023-04-11 22:01:56 +00:00
Andrew FerlitschandGitHub	1d6f9bc36b	fix: exception (#1714 ) * fix: exception * fix: exception * fix: exception * fix: exception	2023-04-11 20:40:22 +00:00
dstnluong-googleandGitHub	973ecf95e7	Fix retinanet_spinenet143 experiment args in IOD notebook to use correct config file. (#1716 ) * Minor nit fixes for model garden tfvision IOD notebook. * Sync * Lint * Fix retinanet_spinenet143 experiment args in IOD notebook to use correct config file.	2023-04-11 20:22:29 +00:00
Xiang XuandGitHub	1f2adec703	fix prediction routes (#1715 )	2023-04-11 19:33:17 +00:00
Xiang XuandGitHub	9843c1f063	fix image url (#1711 )	2023-04-11 16:06:14 +00:00
KCFindstrandGitHub	5afdc3524a	Add a trailing slash to Model Garden ICN notebook checkpoint destination URI (#1710 )	2023-04-11 16:04:58 +00:00
Andrew FerlitschandGitHub	770508b4f0	fix: delete experiment and fs (#1709 ) * fix: delete experiment and fs * fix False typo * fix: 2nd try at Falsee typo * fix: update_time	2023-04-10 21:56:20 +00:00
Xiang XuandGitHub	987881e887	fix links (#1708 )	2023-04-10 14:44:21 +00:00
Andrew FerlitschandGitHub	8e53b623e5	fix: migrate vizier notebook (#1706 ) * fix: migrate vizier notebook * fix: review comment	2023-04-07 06:55:57 +00:00
Andrew FerlitschandGitHub	948537e1d4	fix: migrate machine management (#1707 ) * fix: migrate machine management * remove not per review	2023-04-06 20:57:29 +00:00
Andrew FerlitschandGitHub	4338b1d90b	fix: unique str (#1692 ) * fix: unique str * fix: lint * fix cleanup	2023-04-06 12:43:38 +00:00
Andrew FerlitschandGitHub	b7a41637ea	fix: unique str (#1693 ) * fix: unique str * Fix cleanup	2023-04-06 12:43:03 +00:00
Andrew FerlitschandGitHub	c8f7b910f2	fix: add back missing not (#1687 )	2023-04-05 22:48:40 +00:00
Andrew FerlitschandGitHub	1e226a3419	fix: unique str (#1701 )	2023-04-05 20:38:22 +00:00
Andrew FerlitschandGitHub	287f70abc2	fix: unique dtr (#1700 )	2023-04-05 20:37:37 +00:00
Andrew FerlitschandGitHub	8e9d664d88	fix: unique dtr (#1699 )	2023-04-05 20:36:29 +00:00
Andrew FerlitschandGitHub	166a407b8d	fix: unique str (#1698 )	2023-04-05 20:35:29 +00:00
Andrew FerlitschandGitHub	72f9ba7647	fix: unique str (#1697 )	2023-04-05 20:34:41 +00:00
Andrew FerlitschandGitHub	24549bc506	fix: unique str (#1696 )	2023-04-05 20:33:46 +00:00
Andrew FerlitschandGitHub	dc4efb0c63	fix: unique str (#1694 )	2023-04-05 20:33:04 +00:00
Andrew FerlitschandGitHub	b8c6cc29b4	fix: unique str (#1702 )	2023-04-05 20:31:59 +00:00
Andrew FerlitschandGitHub	3b6d2e02d0	fix: unique str (#1703 )	2023-04-05 20:31:18 +00:00
Andrew FerlitschandGitHub	f33adcea80	fix: fine tune unique (#1704 )	2023-04-05 20:30:51 +00:00
Andrew FerlitschandGitHub	5b6063b8a8	fix: unique str (#1695 )	2023-04-05 18:56:37 +00:00
Andrew FerlitschandGitHub	1857b23556	fix: unique str (#1691 )	2023-04-05 18:55:52 +00:00
Andrew FerlitschandGitHub	86d85ca555	fix: unique str (#1690 )	2023-04-05 18:55:18 +00:00
Andrew FerlitschandGitHub	67b50fda2c	fix: update unique str (#1689 )	2023-04-05 18:54:34 +00:00
dstnluong-googleandGitHub	0d5cb22491	Minor nit fixes for model garden tfvision IOD notebook. (#1684 ) * Minor nit fixes for model garden tfvision IOD notebook. * Sync * Lint	2023-04-05 16:52:32 +00:00