diff --git a/notebooks/community/model_garden/model_garden_pytorch_minimax_m2_deployment.ipynb b/notebooks/community/model_garden/model_garden_pytorch_minimax_m2_deployment.ipynb index 8d38ce1f2..de3e618d8 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_minimax_m2_deployment.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_minimax_m2_deployment.ipynb @@ -237,7 +237,7 @@ }, "outputs": [], "source": [ - "model_version = \"minimax-m2\" # @param [\"minimax-m2\", \"minimax-m2.1\"] {isTemplate:true}\n", + "model_version = \"minimax-m2\" # @param [\"minimax-m2\", \"minimax-m2.1\", \"minimax-m2.5\"] {isTemplate:true}\n", "MODEL_NAME = f\"minimaxai/minimax-m2@{model_version}\"" ] }, @@ -382,7 +382,7 @@ "endpoints[\"sdk_custom\"] = model.deploy(\n", " accept_eula=True,\n", " use_dedicated_endpoint=use_dedicated_endpoint,\n", - " serving_container_image_uri=\"us-docker.pkg.dev/vertex-ai/vertex-vision-model-garden-dockers/pytorch-sglang-serve:sglang-airlock-20251028-1830\",\n", + " serving_container_image_uri=\"us-docker.pkg.dev/vertex-ai/vertex-vision-model-garden-dockers/pytorch-sglang-serve:airlock-v0.5.7\",\n", " machine_type=\"a3-highgpu-8g\",\n", " accelerator_type=\"NVIDIA_H100_80GB\",\n", " accelerator_count=8,\n",