From ecd1fc28d2a9b11097faa6621acef3dfcb407a51 Mon Sep 17 00:00:00 2001
From: Dustin Luong <dustinluong@google.com>
Date: Wed, 30 Apr 2025 21:32:44 -0700
Subject: [PATCH] Set model id to GCS bucket for Qwen3 235B model for better
 stability.

PiperOrigin-RevId: 753430192
---
 .../model_garden/model_garden_pytorch_qwen3_deployment.ipynb     | 1 +
 1 file changed, 1 insertion(+)

diff --git a/notebooks/community/model_garden/model_garden_pytorch_qwen3_deployment.ipynb b/notebooks/community/model_garden/model_garden_pytorch_qwen3_deployment.ipynb
index 3b6e4cb17..b13198f94 100644
--- a/notebooks/community/model_garden/model_garden_pytorch_qwen3_deployment.ipynb
+++ b/notebooks/community/model_garden/model_garden_pytorch_qwen3_deployment.ipynb
@@ -259,6 +259,7 @@
         "    if base_model_name in [\"Qwen3-235B-A22B\"]:\n",
         "        machine_type = \"a3-highgpu-8g\"\n",
         "        accelerator_count = 8\n",
+        "        model_id = \"gs://vertex-model-garden-restricted-us/Qwen/Qwen3-235B-A22B\"\n",
         "    else:\n",
         "        machine_type = \"a3-highgpu-2g\"\n",
         "        accelerator_count = 2\n",