Skip to content

Commit 75eb777

Browse files
Ark-kuncopybara-github
authored andcommitted
chore: LLM - Change tuning to use "@" instead of "-" as the model version delimiter
The "<model>-<version>" format was only supported for text-bison-001 and will be deprecated in the future. PiperOrigin-RevId: 550710642
1 parent fcf4ac9 commit 75eb777

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

tests/unit/aiplatform/test_language_models.py

+1
Original file line numberDiff line numberDiff line change
@@ -672,6 +672,7 @@ def test_tune_model(
672672
"pipeline_job"
673673
].runtime_config.parameter_values
674674
assert pipeline_arguments["learning_rate"] == 0.1
675+
assert pipeline_arguments["large_model_reference"] == "text-bison@001"
675676
assert (
676677
call_kwargs["pipeline_job"].encryption_spec.kms_key_name
677678
== _TEST_ENCRYPTION_KEY_NAME

vertexai/_model_garden/_model_garden_models.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def _get_model_info(
117117

118118
if short_model_id in _SHORT_MODEL_ID_TO_TUNING_PIPELINE_MAP:
119119
tuning_pipeline_uri = _SHORT_MODEL_ID_TO_TUNING_PIPELINE_MAP[short_model_id]
120-
tuning_model_id = short_model_id + "-" + publisher_model_res.version_id
120+
tuning_model_id = publisher_model_template.rsplit("/", 1)[-1]
121121
else:
122122
tuning_pipeline_uri = None
123123
tuning_model_id = None

0 commit comments

Comments
 (0)