From 3a776a7a41d7b0e8f4a32e5e50401c33bde06985 Mon Sep 17 00:00:00 2001 From: A Vertex SDK engineer Date: Wed, 23 Jul 2025 11:23:59 -0700 Subject: [PATCH] feat: Allow adapter_size=32 for supervised tuning PiperOrigin-RevId: 786356415 --- vertexai/preview/tuning/_supervised_tuning.py | 8 ++++++-- vertexai/tuning/_supervised_tuning.py | 8 ++++++-- 2 files changed, 12 insertions(+), 4 deletions(-) diff --git a/vertexai/preview/tuning/_supervised_tuning.py b/vertexai/preview/tuning/_supervised_tuning.py index da0596880d..7c90f20f6d 100644 --- a/vertexai/preview/tuning/_supervised_tuning.py +++ b/vertexai/preview/tuning/_supervised_tuning.py @@ -39,7 +39,7 @@ def train( epochs: Optional[int] = None, learning_rate: Optional[float] = None, learning_rate_multiplier: Optional[float] = None, - adapter_size: Optional[Literal[1, 4, 8, 16]] = None, + adapter_size: Optional[Literal[1, 4, 8, 16, 32]] = None, labels: Optional[Dict[str, str]] = None, output_uri: Optional[str] = None, ) -> "SupervisedTuningJob": @@ -103,9 +103,13 @@ def train( adapter_size_value = ( gca_tuning_job_types.SupervisedHyperParameters.AdapterSize.ADAPTER_SIZE_SIXTEEN ) + elif adapter_size == 32: + adapter_size_value = ( + gca_tuning_job_types.SupervisedHyperParameters.AdapterSize.ADAPTER_SIZE_THIRTY_TWO + ) else: raise ValueError( - f"Unsupported adapter size: {adapter_size}. The supported sizes are [1, 4, 8, 16]" + f"Unsupported adapter size: {adapter_size}. The supported sizes are [1, 4, 8, 16, 32]" ) if isinstance(train_dataset, datasets.MultimodalDataset): train_dataset = train_dataset.resource_name diff --git a/vertexai/tuning/_supervised_tuning.py b/vertexai/tuning/_supervised_tuning.py index 7f3f8616c5..3801fdb470 100644 --- a/vertexai/tuning/_supervised_tuning.py +++ b/vertexai/tuning/_supervised_tuning.py @@ -32,7 +32,7 @@ def train( tuned_model_display_name: Optional[str] = None, epochs: Optional[int] = None, learning_rate_multiplier: Optional[float] = None, - adapter_size: Optional[Literal[1, 4, 8, 16]] = None, + adapter_size: Optional[Literal[1, 4, 8, 16, 32]] = None, labels: Optional[Dict[str, str]] = None, ) -> "SupervisedTuningJob": """Tunes a model using supervised training. @@ -70,9 +70,13 @@ def train( adapter_size_value = ( gca_tuning_job_types.SupervisedHyperParameters.AdapterSize.ADAPTER_SIZE_SIXTEEN ) + elif adapter_size == 32: + adapter_size_value = ( + gca_tuning_job_types.SupervisedHyperParameters.AdapterSize.ADAPTER_SIZE_THIRTY_TWO + ) else: raise ValueError( - f"Unsupported adapter size: {adapter_size}. The supported sizes are [1, 4, 8, 16]" + f"Unsupported adapter size: {adapter_size}. The supported sizes are [1, 4, 8, 16, 32]" ) if isinstance(train_dataset, datasets.MultimodalDataset): train_dataset = train_dataset.resource_name