Skip to content

Commit bae8429

Browse files
Ark-kuncopybara-github
authored andcommitted
feat: LLM - Made the tuning location parameters truly optional
PiperOrigin-RevId: 631312780
1 parent 88188d2 commit bae8429

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

vertexai/language_models/_language_models.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -283,6 +283,11 @@ def tune_model(
283283
ValueError: If the "tuned_model_location" value is not supported
284284
RuntimeError: If the model does not support tuning
285285
"""
286+
if tuning_job_location is None:
287+
tuning_job_location = aiplatform_initializer.global_config.location
288+
if tuned_model_location is None:
289+
tuned_model_location = aiplatform_initializer.global_config.location
290+
286291
tuning_parameters = {}
287292
if batch_size is not None:
288293
tuning_parameters["batch_size"] = batch_size
@@ -623,7 +628,6 @@ def _tune_model_rlhf(
623628
624629
Args:
625630
tuning_parameters: Tuning pipeline parameter values.
626-
tuning_job_location: GCP location where the tuning job should be run.
627631
628632
Returns:
629633
A `LanguageModelTuningJob` object that represents the tuning job.

0 commit comments

Comments
 (0)