Allows the user to select spot VM for deployment

vertex-mg-bot · copybara-github · commit 2f5119a266cd · 2025-12-15T21:23:11.000-08:00
PiperOrigin-RevId: 845064375
diff --git a/notebooks/community/model_garden/model_garden_gemma2_finetuning_on_vertex.ipynb b/notebooks/community/model_garden/model_garden_gemma2_finetuning_on_vertex.ipynb
@@ -109,7 +109,8 @@
         "! pip install --upgrade --quiet gcsfs==2024.3.1\n",
         "! pip install --upgrade --quiet accelerate==0.34.2\n",
         "! pip install --upgrade --quiet transformers==4.47.1\n",
-        "! pip install --upgrade --quiet datasets==2.20.0"
+        "! pip install --upgrade --quiet datasets==2.20.0\n",
+        "! pip install --upgrade --quiet google-cloud-aiplatform==1.130.0"
       ]
     },
     {
diff --git a/notebooks/community/model_garden/model_garden_gemma_finetuning_on_vertex.ipynb b/notebooks/community/model_garden/model_garden_gemma_finetuning_on_vertex.ipynb
@@ -97,7 +97,7 @@
         "# @title Install Python Packages for Finetuning\n",
         "\n",
         "# @markdown 1. Install google-cloud-aiplatform package and restart the session if instructed.\n",
-        "! pip install --upgrade --quiet 'google-cloud-aiplatform>=1.66.0'\n",
+        "! pip install --upgrade --quiet google-cloud-aiplatform==1.130.0\n",
         "\n",
         "# @markdown 2. Install packages to validate dataset with template.\n",
         "! pip install --upgrade --quiet accelerate==0.31.0\n",
diff --git a/notebooks/community/model_garden/model_garden_llama3_1_finetuning_with_workbench.ipynb b/notebooks/community/model_garden/model_garden_llama3_1_finetuning_with_workbench.ipynb
@@ -148,7 +148,8 @@
         "! pip install --quiet accelerate==0.31.0\n",
         "! pip install --quiet transformers==4.43.1\n",
         "! pip install --quiet datasets==2.19.2\n",
-        "! pip install --quiet tensorflow==2.18.0"
+        "! pip install --quiet tensorflow==2.18.0\n",
+        "! pip install --upgrade --quiet google-cloud-aiplatform==1.130.0"
       ]
     },
     {
diff --git a/notebooks/community/model_garden/model_garden_pytorch_gemma_peft_finetuning_hf.ipynb b/notebooks/community/model_garden/model_garden_pytorch_gemma_peft_finetuning_hf.ipynb
@@ -97,7 +97,7 @@
         "# @title Install Python Packages for Finetuning\n",
         "\n",
         "# @markdown 1. Install google-cloud-aiplatform package and restart the session if instructed.\n",
-        "! pip install --upgrade --quiet 'google-cloud-aiplatform>=1.66.0'\n",
+        "! pip install --upgrade --quiet google-cloud-aiplatform==1.130.0\n",
         "\n",
         "# @markdown 2. Install packages to validate dataset with template.\n",
         "! pip install --upgrade --quiet accelerate==0.31.0\n",
diff --git a/notebooks/community/model_garden/model_garden_pytorch_llama3_1_finetuning.ipynb b/notebooks/community/model_garden/model_garden_pytorch_llama3_1_finetuning.ipynb
@@ -104,7 +104,8 @@
         "! pip install --upgrade --quiet gcsfs==2024.3.1\n",
         "! pip install --upgrade --quiet accelerate==0.34.2\n",
         "! pip install --upgrade --quiet transformers==4.47.1\n",
-        "! pip install --upgrade --quiet datasets==2.20.0"
+        "! pip install --upgrade --quiet datasets==2.20.0\n",
+        "! pip install --upgrade --quiet google-cloud-aiplatform==1.130.0"
       ]
     },
     {
diff --git a/notebooks/community/model_garden/model_garden_pytorch_llama3_3_finetuning.ipynb b/notebooks/community/model_garden/model_garden_pytorch_llama3_3_finetuning.ipynb
@@ -104,7 +104,8 @@
         "! pip install --upgrade --quiet gcsfs==2024.3.1\n",
         "! pip install --upgrade --quiet accelerate==0.34.2\n",
         "! pip install --upgrade --quiet transformers==4.47.1\n",
-        "! pip install --upgrade --quiet datasets==2.20.0"
+        "! pip install --upgrade --quiet datasets==2.20.0\n",
+        "! pip install --upgrade --quiet google-cloud-aiplatform==1.130.0"
       ]
     },
     {
diff --git a/notebooks/community/model_garden/model_garden_pytorch_mistral_peft_tuning.ipynb b/notebooks/community/model_garden/model_garden_pytorch_mistral_peft_tuning.ipynb
@@ -97,7 +97,7 @@
         "# @title Install Python Packages for Finetuning\n",
         "\n",
         "# @markdown 1. Install google-cloud-aiplatform package and restart the session if instructed.\n",
-        "! pip install --upgrade --quiet 'google-cloud-aiplatform>=1.66.0'\n",
+        "! pip install --upgrade --quiet google-cloud-aiplatform==1.130.0\n",
         "\n",
         "# @markdown 2. Install packages to validate dataset with template.\n",
         "! pip install --upgrade --quiet gcsfs==2024.3.1\n",
diff --git a/notebooks/community/model_garden/model_garden_pytorch_mixtral_peft_tuning.ipynb b/notebooks/community/model_garden/model_garden_pytorch_mixtral_peft_tuning.ipynb
@@ -97,7 +97,7 @@
         "# @title Install Python Packages for Finetuning\n",
         "\n",
         "# @markdown 1. Install google-cloud-aiplatform package and restart the session if instructed.\n",
-        "! pip install --upgrade --quiet 'google-cloud-aiplatform>=1.66.0'\n",
+        "! pip install --upgrade --quiet google-cloud-aiplatform==1.130.0\n",
         "\n",
         "# @markdown 2. Install packages to validate dataset with template.\n",
         "! pip install --upgrade --quiet gcsfs==2024.3.1\n",

Original file line number	Diff line number	Diff line change
`@@ -109,7 +109,8 @@`
`109`	`109`	`"! pip install --upgrade --quiet gcsfs==2024.3.1\n",`
`110`	`110`	`"! pip install --upgrade --quiet accelerate==0.34.2\n",`
`111`	`111`	`"! pip install --upgrade --quiet transformers==4.47.1\n",`
`112`		`- "! pip install --upgrade --quiet datasets==2.20.0"`
	`112`	`+ "! pip install --upgrade --quiet datasets==2.20.0\n",`
	`113`	`+ "! pip install --upgrade --quiet google-cloud-aiplatform==1.130.0"`
`113`	`114`	`]`
`114`	`115`	`},`
`115`	`116`	`{`
Original file line number	Diff line number	Diff line change
`@@ -148,7 +148,8 @@`
`148`	`148`	`"! pip install --quiet accelerate==0.31.0\n",`
`149`	`149`	`"! pip install --quiet transformers==4.43.1\n",`
`150`	`150`	`"! pip install --quiet datasets==2.19.2\n",`
`151`		`- "! pip install --quiet tensorflow==2.18.0"`
	`151`	`+ "! pip install --quiet tensorflow==2.18.0\n",`
	`152`	`+ "! pip install --upgrade --quiet google-cloud-aiplatform==1.130.0"`
`152`	`153`	`]`
`153`	`154`	`},`
`154`	`155`	`{`
Original file line number	Diff line number	Diff line change
`@@ -104,7 +104,8 @@`
`104`	`104`	`"! pip install --upgrade --quiet gcsfs==2024.3.1\n",`
`105`	`105`	`"! pip install --upgrade --quiet accelerate==0.34.2\n",`
`106`	`106`	`"! pip install --upgrade --quiet transformers==4.47.1\n",`
`107`		`- "! pip install --upgrade --quiet datasets==2.20.0"`
	`107`	`+ "! pip install --upgrade --quiet datasets==2.20.0\n",`
	`108`	`+ "! pip install --upgrade --quiet google-cloud-aiplatform==1.130.0"`
`108`	`109`	`]`
`109`	`110`	`},`
`110`	`111`	`{`