Skip to content

Commit 9eba18f

Browse files
Ark-kuncopybara-github
authored andcommitted
feat: LLM - Added tuning support for the *-bison-32k models
PiperOrigin-RevId: 579975371
1 parent 35925c0 commit 9eba18f

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

vertexai/_model_garden/_model_garden_models.py

+4
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,13 @@
3535

3636
_SHORT_MODEL_ID_TO_TUNING_PIPELINE_MAP = {
3737
"text-bison": "https://us-kfp.pkg.dev/ml-pipeline/large-language-model-pipelines/tune-large-model/v2.0.0",
38+
"text-bison-32k": "https://us-kfp.pkg.dev/ml-pipeline/large-language-model-pipelines/tune-large-model/v2.0.0",
3839
"code-bison": "https://us-kfp.pkg.dev/ml-pipeline/large-language-model-pipelines/tune-large-model/v3.0.0",
40+
"code-bison-32k": "https://us-kfp.pkg.dev/ml-pipeline/large-language-model-pipelines/tune-large-model/v3.0.0",
3941
"chat-bison": "https://us-kfp.pkg.dev/ml-pipeline/large-language-model-pipelines/tune-large-chat-model/v3.0.0",
42+
"chat-bison-32k": "https://us-kfp.pkg.dev/ml-pipeline/large-language-model-pipelines/tune-large-chat-model/v3.0.0",
4043
"codechat-bison": "https://us-kfp.pkg.dev/ml-pipeline/large-language-model-pipelines/tune-large-chat-model/v3.0.0",
44+
"codechat-bison-32k": "https://us-kfp.pkg.dev/ml-pipeline/large-language-model-pipelines/tune-large-chat-model/v3.0.0",
4145
}
4246

4347
_SDK_PRIVATE_PREVIEW_LAUNCH_STAGE = frozenset(

0 commit comments

Comments
 (0)