Class DistillationSpec.Builder
-
- All Implemented Interfaces:
public abstract class DistillationSpec.BuilderBuilder for DistillationSpec.
-
-
Constructor Summary
Constructors Constructor Description DistillationSpec.Builder()
-
Method Summary
Modifier and Type Method Description abstract DistillationSpec.BuilderpromptDatasetUri(String promptDatasetUri)Setter for promptDatasetUri. DistillationSpec.BuilderclearPromptDatasetUri()Clears the value of promptDatasetUri field. abstract DistillationSpec.BuilderbaseTeacherModel(String baseTeacherModel)Setter for baseTeacherModel. DistillationSpec.BuilderclearBaseTeacherModel()Clears the value of baseTeacherModel field. abstract DistillationSpec.BuilderhyperParameters(DistillationHyperParameters hyperParameters)Setter for hyperParameters. DistillationSpec.BuilderhyperParameters(DistillationHyperParameters.Builder hyperParametersBuilder)Setter for hyperParameters builder. DistillationSpec.BuilderclearHyperParameters()Clears the value of hyperParameters field. abstract DistillationSpec.BuilderpipelineRootDirectory(String pipelineRootDirectory)Setter for pipelineRootDirectory. DistillationSpec.BuilderclearPipelineRootDirectory()Clears the value of pipelineRootDirectory field. abstract DistillationSpec.BuilderstudentModel(String studentModel)Setter for studentModel. DistillationSpec.BuilderclearStudentModel()Clears the value of studentModel field. abstract DistillationSpec.BuildertrainingDatasetUri(String trainingDatasetUri)Setter for trainingDatasetUri. DistillationSpec.BuilderclearTrainingDatasetUri()Clears the value of trainingDatasetUri field. abstract DistillationSpec.BuildertunedTeacherModelSource(String tunedTeacherModelSource)Setter for tunedTeacherModelSource. DistillationSpec.BuilderclearTunedTeacherModelSource()Clears the value of tunedTeacherModelSource field. abstract DistillationSpec.BuildervalidationDatasetUri(String validationDatasetUri)Setter for validationDatasetUri. DistillationSpec.BuilderclearValidationDatasetUri()Clears the value of validationDatasetUri field. abstract DistillationSpec.BuildertuningMode(TuningMode tuningMode)Setter for tuningMode. DistillationSpec.BuilderclearTuningMode()Clears the value of tuningMode field. DistillationSpec.BuildertuningMode(TuningMode.Known knownType)Setter for tuningMode given a known enum. DistillationSpec.BuildertuningMode(String tuningMode)Setter for tuningMode given a string. abstract DistillationSpecbuild()-
-
Method Detail
-
promptDatasetUri
abstract DistillationSpec.Builder promptDatasetUri(String promptDatasetUri)
Setter for promptDatasetUri.
promptDatasetUri: The GCS URI of the prompt dataset to use during distillation.
-
clearPromptDatasetUri
@CanIgnoreReturnValue() DistillationSpec.Builder clearPromptDatasetUri()
Clears the value of promptDatasetUri field.
-
baseTeacherModel
abstract DistillationSpec.Builder baseTeacherModel(String baseTeacherModel)
Setter for baseTeacherModel.
baseTeacherModel: The base teacher model that is being distilled. See [Supported models](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/tuning#supported_models).
-
clearBaseTeacherModel
@CanIgnoreReturnValue() DistillationSpec.Builder clearBaseTeacherModel()
Clears the value of baseTeacherModel field.
-
hyperParameters
abstract DistillationSpec.Builder hyperParameters(DistillationHyperParameters hyperParameters)
Setter for hyperParameters.
hyperParameters: Optional. Hyperparameters for Distillation.
-
hyperParameters
@CanIgnoreReturnValue() DistillationSpec.Builder hyperParameters(DistillationHyperParameters.Builder hyperParametersBuilder)
Setter for hyperParameters builder.
hyperParameters: Optional. Hyperparameters for Distillation.
-
clearHyperParameters
@CanIgnoreReturnValue() DistillationSpec.Builder clearHyperParameters()
Clears the value of hyperParameters field.
-
pipelineRootDirectory
abstract DistillationSpec.Builder pipelineRootDirectory(String pipelineRootDirectory)
Setter for pipelineRootDirectory.
pipelineRootDirectory: Deprecated. A path in a Cloud Storage bucket, which will be treated as the root output directory of the distillation pipeline. It is used by the system to generate the paths of output artifacts.
-
clearPipelineRootDirectory
@CanIgnoreReturnValue() DistillationSpec.Builder clearPipelineRootDirectory()
Clears the value of pipelineRootDirectory field.
-
studentModel
abstract DistillationSpec.Builder studentModel(String studentModel)
Setter for studentModel.
studentModel: The student model that is being tuned, e.g., "google/gemma-2b-1.1-it". Deprecated. Use base_model instead.
-
clearStudentModel
@CanIgnoreReturnValue() DistillationSpec.Builder clearStudentModel()
Clears the value of studentModel field.
-
trainingDatasetUri
abstract DistillationSpec.Builder trainingDatasetUri(String trainingDatasetUri)
Setter for trainingDatasetUri.
trainingDatasetUri: Deprecated. Cloud Storage path to file containing training dataset for tuning. The dataset must be formatted as a JSONL file.
-
clearTrainingDatasetUri
@CanIgnoreReturnValue() DistillationSpec.Builder clearTrainingDatasetUri()
Clears the value of trainingDatasetUri field.
-
tunedTeacherModelSource
abstract DistillationSpec.Builder tunedTeacherModelSource(String tunedTeacherModelSource)
Setter for tunedTeacherModelSource.
tunedTeacherModelSource: The resource name of the Tuned teacher model. Format: `projects/{project}/locations/{location}/models/{model}`.
-
clearTunedTeacherModelSource
@CanIgnoreReturnValue() DistillationSpec.Builder clearTunedTeacherModelSource()
Clears the value of tunedTeacherModelSource field.
-
validationDatasetUri
abstract DistillationSpec.Builder validationDatasetUri(String validationDatasetUri)
Setter for validationDatasetUri.
validationDatasetUri: Optional. Cloud Storage path to file containing validation dataset for tuning. The dataset must be formatted as a JSONL file.
-
clearValidationDatasetUri
@CanIgnoreReturnValue() DistillationSpec.Builder clearValidationDatasetUri()
Clears the value of validationDatasetUri field.
-
tuningMode
abstract DistillationSpec.Builder tuningMode(TuningMode tuningMode)
Setter for tuningMode.
tuningMode: Tuning mode for tuning.
-
clearTuningMode
@CanIgnoreReturnValue() DistillationSpec.Builder clearTuningMode()
Clears the value of tuningMode field.
-
tuningMode
@CanIgnoreReturnValue() DistillationSpec.Builder tuningMode(TuningMode.Known knownType)
Setter for tuningMode given a known enum.
tuningMode: Tuning mode for tuning.
-
tuningMode
@CanIgnoreReturnValue() DistillationSpec.Builder tuningMode(String tuningMode)
Setter for tuningMode given a string.
tuningMode: Tuning mode for tuning.
-
build
abstract DistillationSpec build()
-
-
-
-