File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -283,6 +283,11 @@ def tune_model(
283
283
ValueError: If the "tuned_model_location" value is not supported
284
284
RuntimeError: If the model does not support tuning
285
285
"""
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
+
286
291
tuning_parameters = {}
287
292
if batch_size is not None :
288
293
tuning_parameters ["batch_size" ] = batch_size
@@ -623,7 +628,6 @@ def _tune_model_rlhf(
623
628
624
629
Args:
625
630
tuning_parameters: Tuning pipeline parameter values.
626
- tuning_job_location: GCP location where the tuning job should be run.
627
631
628
632
Returns:
629
633
A `LanguageModelTuningJob` object that represents the tuning job.
You can’t perform that action at this time.
0 commit comments