chore: Support new tuning algorithm and pipeline

Ark-kun · copybara-github · commit a4bd9bd292a3 · 2023-07-12T04:11:19.000-07:00
PiperOrigin-RevId: 547454144
diff --git a/vertexai/_model_garden/_model_garden_models.py b/vertexai/_model_garden/_model_garden_models.py
@@ -31,7 +31,7 @@
 _SUPPORTED_PUBLISHERS = ["google"]
 
 _SHORT_MODEL_ID_TO_TUNING_PIPELINE_MAP = {
-    "text-bison": "https://us-kfp.pkg.dev/vertex-ai/large-language-model-pipelines/tune-large-model/sdk-1-25"
+    "text-bison": "https://us-kfp.pkg.dev/ml-pipeline/large-language-model-pipelines/tune-large-model/v2.0.0"
 }
 
 _SDK_PRIVATE_PREVIEW_LAUNCH_STAGE = frozenset(

Original file line number	Diff line number	Diff line change
`@@ -31,7 +31,7 @@`
`31`	`31`	`_SUPPORTED_PUBLISHERS = ["google"]`
`32`	`32`
`33`	`33`	`_SHORT_MODEL_ID_TO_TUNING_PIPELINE_MAP = {`
`34`		`- "text-bison": "https://us-kfp.pkg.dev/vertex-ai/large-language-model-pipelines/tune-large-model/sdk-1-25"`
	`34`	`+ "text-bison": "https://us-kfp.pkg.dev/ml-pipeline/large-language-model-pipelines/tune-large-model/v2.0.0"`
`35`	`35`	`}`
`36`	`36`
`37`	`37`	`_SDK_PRIVATE_PREVIEW_LAUNCH_STAGE = frozenset(`