Class DistillationSpec.Builder
- Enclosing class:
- DistillationSpec
-
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptionabstract DistillationSpec.BuilderbaseTeacherModel(String baseTeacherModel) Setter for baseTeacherModel.abstract DistillationSpecbuild()Clears the value of baseTeacherModel field.Clears the value of hyperParameters field.Clears the value of pipelineRootDirectory field.Clears the value of promptDatasetUri field.Clears the value of studentModel field.Clears the value of trainingDatasetUri field.Clears the value of tunedTeacherModelSource field.Clears the value of validationDatasetUri field.abstract DistillationSpec.BuilderhyperParameters(DistillationHyperParameters hyperParameters) Setter for hyperParameters.hyperParameters(DistillationHyperParameters.Builder hyperParametersBuilder) Setter for hyperParameters builder.abstract DistillationSpec.BuilderpipelineRootDirectory(String pipelineRootDirectory) Setter for pipelineRootDirectory.abstract DistillationSpec.BuilderpromptDatasetUri(String promptDatasetUri) Setter for promptDatasetUri.abstract DistillationSpec.BuilderstudentModel(String studentModel) Setter for studentModel.abstract DistillationSpec.BuildertrainingDatasetUri(String trainingDatasetUri) Setter for trainingDatasetUri.abstract DistillationSpec.BuildertunedTeacherModelSource(String tunedTeacherModelSource) Setter for tunedTeacherModelSource.abstract DistillationSpec.BuildervalidationDatasetUri(String validationDatasetUri) Setter for validationDatasetUri.
-
Constructor Details
-
Builder
public Builder()
-
-
Method Details
-
promptDatasetUri
Setter for promptDatasetUri.promptDatasetUri: The GCS URI of the prompt dataset to use during distillation.
-
clearPromptDatasetUri
Clears the value of promptDatasetUri field. -
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
Clears the value of baseTeacherModel field. -
hyperParameters
public abstract DistillationSpec.Builder hyperParameters(DistillationHyperParameters hyperParameters) Setter for hyperParameters.hyperParameters: Optional. Hyperparameters for Distillation.
-
hyperParameters
@CanIgnoreReturnValue public DistillationSpec.Builder hyperParameters(DistillationHyperParameters.Builder hyperParametersBuilder) Setter for hyperParameters builder.hyperParameters: Optional. Hyperparameters for Distillation.
-
clearHyperParameters
Clears the value of hyperParameters field. -
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
Clears the value of pipelineRootDirectory field. -
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
Clears the value of studentModel field. -
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
Clears the value of trainingDatasetUri field. -
tunedTeacherModelSource
Setter for tunedTeacherModelSource.tunedTeacherModelSource: The resource name of the Tuned teacher model. Format: `projects/{project}/locations/{location}/models/{model}`.
-
clearTunedTeacherModelSource
Clears the value of tunedTeacherModelSource field. -
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
Clears the value of validationDatasetUri field. -
build
-