fix: positional params

* Fix local inference when loading weights from GCS

* remove extra <td>

notebooks/community/model_garden/model_garden_keras_stable_diffusion.ipynb

@@ -44,7 +44,7 @@
         "      View on GitHub\n",
         "    </a>\n",
         "  </td>\n",
-        "  <td>                                                                                               <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_keras_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",
@@ -282,8 +282,11 @@
         "from io import BytesIO\n",
         "\n",
         "import matplotlib.pyplot as plt\n",
+        "from google.cloud import storage\n",
         "from PIL import Image\n",
         "\n",
+        "GCS_URI_PREFIX = \"gs://\"\n",
+        "\n",
         "# Training constants.\n",
         "TRAINING_JOB_PREFIX = \"train\"\n",
         "TRAIN_CONTAINER_URI = \"us-docker.pkg.dev/vertex-ai-restricted/vertex-vision-model-garden-dockers/keras-train:latest\"\n",
@@ -317,6 +320,21 @@
         "    return gcs_path\n",
         "\n",
         "\n",
+        "def download_gcs_file_to_local(gcs_uri: str, local_path: str):\n",
+        "    \"\"\"Download a gcs file to a local path.\n",
+        "\n",
+        "    Args:\n",
+        "      gcs_uri: A string of file path on GCS.\n",
+        "      local_path: A string of local file path.\n",
+        "    \"\"\"\n",
+        "    if not gcs_uri.startswith(GCS_URI_PREFIX):\n",
+        "        raise ValueError(f\"{gcs_uri} is not a GCS path starting with {GCS_URI_PREFIX}.\")\n",
+        "    client = storage.Client()\n",
+        "    os.makedirs(os.path.dirname(local_path), exist_ok=True)\n",
+        "    with open(local_path, \"wb\") as f:\n",
+        "        client.download_blob_to_file(gcs_uri, f)\n",
+        "\n",
+        "\n",
         "def deploy_model(model_path, service_account):\n",
         "\n",
         "    deploy_model_name = get_job_name_with_datetime(DEPLOY_JOB_PREFIX)\n",
@@ -420,7 +438,11 @@
         "from keras_cv.models import StableDiffusion\n",
         "\n",
         "model = StableDiffusion(img_height=RESOLUTION, img_width=RESOLUTION, jit_compile=True)\n",
-        "if model_path:\n",
+        "if model_path.startswith(GCS_URI_PREFIX):\n",
+        "    local_model_path = \"/tmp/saved_model.h5\"\n",
+        "    download_gcs_file_to_local(model_path, local_model_path)\n",
+        "    model.diffusion_model.load_weights(local_model_path)\n",
+        "elif model_path:\n",
         "    model.diffusion_model.load_weights(model_path)"
       ]
     },
@@ -568,7 +590,7 @@
       },
       "source": [
         "## Finetune models\n",
-        "This section shows how to finetune Keras Stable diffusion models with trainig dockers.\n",
+        "This section shows how to finetune Keras Stable diffusion models with training dockers.\n",
         "\n",
         "If you would like to use finetuned models, please go to the section `Run inferences`."
       ]

notebooks/community/model_garden/model_garden_proprietary_image_classification.ipynb

@@ -389,7 +389,6 @@
         "- `model_type`: The type of model for deployment.\n",
         "  - `EFFICIENTNET`: A model that is available in Vertex Model Garden image classification training with customizable hyperparameters. Best tailored to be used within Google Cloud, and cannot be exported externally.\n",
         "  - `MAXVIT`: A model that is available in Vertex Model Garden image classification training with customizable hyperparameters. Best tailored to be used within Google Cloud, and cannot be exported externally.\n",
-        "  - `COCA`: A model that is available in Vertex Model Garden image classification training with customizable hyperparameters. Best tailored to be used within Google Cloud, and cannot be exported externally.\n",
         "- `checkpoint_name`: Optional. The field is reserved for Model Garden model training, based on the provided pre-trained model checkpoint.\n",
         "- `trainer_config`: Optional. The field is usually used together with the Model Garden model training when passing the customized configs for the trainer.\n",
         "\n",
@@ -457,7 +456,7 @@
         "METRIC_SPEC_VALUE = \"maximize\"\n",
         "SEARCH_ALGORITHM = \"random\"\n",
         "MEASUREMENT_SELECTION = \"best\"\n",
-        "MODEL_TYPE = \"COCA\"  # @param {type:\"string\"} one of the values [\"COCA\", \"MAXVIT\", \"EFFICIENTNET\"]\n",
+        "MODEL_TYPE = \"MAXVIT\"  # @param {type:\"string\"} one of the values [\"MAXVIT\", \"EFFICIENTNET\"]\n",
         "\n",
         "job = aiplatform.AutoMLImageTrainingJob(\n",
         "    display_name=get_job_name_with_datetime(TRAINING_JOB_PREFIX),\n",

notebooks/official/experiments/comparing_pipeline_runs.ipynb

@@ -714,7 +714,12 @@
         "):\n",
         "\n",
         "    custom_trainer(\n",
-        "        train_uri, label_uri, max_depth, learning_rate, boost_rounds, model_uri\n",
+        "        train_uri=train_uri,\n",
+        "        label_uri=label_uri,\n",
+        "        max_depth=max_depth,\n",
+        "        learning_rate=learning_rate,\n",
+        "        boost_rounds=boost_rounds,\n",
+        "        model_uri=model_uri,\n",
         "    )"
       ]
     },