Skip to content

Commit 939ba3b

Browse files
Google APIscopybara-github
authored andcommitted
feat: Support VeoLoraTuningSpec in the tuning jobs
PiperOrigin-RevId: 903331273
1 parent a787482 commit 939ba3b

1 file changed

Lines changed: 64 additions & 0 deletions

File tree

‎google/cloud/aiplatform/v1beta1/tuning_job.proto‎

Lines changed: 64 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,9 @@ message TuningJob {
6464

6565
// Tuning Spec for Veo Tuning.
6666
VeoTuningSpec veo_tuning_spec = 33;
67+
68+
// Tuning Spec for Veo LoRA Tuning.
69+
VeoLoraTuningSpec veo_lora_tuning_spec = 38;
6770
}
6871

6972
// Output only. Identifier. Resource name of a TuningJob. Format:
@@ -604,6 +607,37 @@ message VeoHyperParameters {
604607

605608
// Tuning task for text to video.
606609
TUNING_TASK_T2V = 2;
610+
611+
// Tuning task for reference to video.
612+
TUNING_TASK_R2V = 3;
613+
}
614+
615+
// The speed of the tuning job. Only supported for Veo 3.0 models.
616+
enum TuningSpeed {
617+
// The default / unset value. For Veo 3.0 models, this defaults to FAST.
618+
TUNING_SPEED_UNSPECIFIED = 0;
619+
620+
// Regular tuning speed.
621+
REGULAR = 1;
622+
623+
// Fast tuning speed.
624+
FAST = 2;
625+
}
626+
627+
// Adapter size for LoRA tuning.
628+
enum AdapterSize {
629+
// Adapter size is unspecified.
630+
ADAPTER_SIZE_UNSPECIFIED = 0;
631+
632+
// Adapter size 8.
633+
// This is the default adapter size for Veo LoRA tuning.
634+
ADAPTER_SIZE_EIGHT = 8;
635+
636+
// Adapter size 16.
637+
ADAPTER_SIZE_SIXTEEN = 16;
638+
639+
// Adapter size 32.
640+
ADAPTER_SIZE_THIRTY_TWO = 32;
607641
}
608642

609643
// Optional. Number of complete passes the model makes over the entire
@@ -615,6 +649,19 @@ message VeoHyperParameters {
615649

616650
// Optional. The tuning task. Either I2V or T2V.
617651
TuningTask tuning_task = 3 [(google.api.field_behavior) = OPTIONAL];
652+
653+
// Optional. The ratio of Google internal dataset to use in the training
654+
// mixture, in range of `[0, 1)`. If `0.2`, it means 20% of Google internal
655+
// dataset and 80% of user dataset will be used for training. If not set, the
656+
// default value is 0.1.
657+
optional double veo_data_mixture_ratio = 4
658+
[(google.api.field_behavior) = OPTIONAL];
659+
660+
// The speed of the tuning job. Only supported for Veo 3.0 models.
661+
optional TuningSpeed tuning_speed = 5;
662+
663+
// Optional. The adapter size for LoRA tuning.
664+
AdapterSize adapter_size = 6 [(google.api.field_behavior) = OPTIONAL];
618665
}
619666

620667
// Tuning Spec for Veo Model Tuning.
@@ -634,6 +681,23 @@ message VeoTuningSpec {
634681
[(google.api.field_behavior) = OPTIONAL];
635682
}
636683

684+
// Tuning Spec for Veo LoRA Model Tuning.
685+
message VeoLoraTuningSpec {
686+
// Required. Training dataset used for tuning. The dataset can be specified as
687+
// either a Cloud Storage path to a JSONL file or as the resource name of a
688+
// Vertex Multimodal Dataset.
689+
string training_dataset_uri = 1 [(google.api.field_behavior) = REQUIRED];
690+
691+
// Optional. Validation dataset used for tuning. The dataset can be specified
692+
// as either a Cloud Storage path to a JSONL file or as the resource name of a
693+
// Vertex Multimodal Dataset.
694+
string validation_dataset_uri = 2 [(google.api.field_behavior) = OPTIONAL];
695+
696+
// Optional. Hyperparameters for Veo LoRA.
697+
VeoHyperParameters hyper_parameters = 3
698+
[(google.api.field_behavior) = OPTIONAL];
699+
}
700+
637701
// Evaluation Config for Tuning Job.
638702
message EvaluationConfig {
639703
// Required. The metrics used for evaluation.

0 commit comments

Comments
 (0)