Set model id to GCS bucket for Qwen3 235B model for better stability.

PiperOrigin-RevId: 753430192
This commit is contained in:
Dustin Luong
2025-04-30 21:33:20 -07:00
committed by Copybara-Service
parent da0874682f
commit ecd1fc28d2
@@ -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",