feat: add the VeoTuningSpec

Google APIs · copybara-github · commit ec771cb34976 · 2025-07-29T11:26:55.000-07:00
PiperOrigin-RevId: 788536453
diff --git a/google/cloud/aiplatform/v1beta1/tuning_job.proto b/google/cloud/aiplatform/v1beta1/tuning_job.proto
@@ -57,6 +57,9 @@ message TuningJob {
 
     // Tuning Spec for open sourced and third party Partner models.
     PartnerModelTuningSpec partner_model_tuning_spec = 21;
+
+    // Tuning Spec for Veo Tuning.
+    VeoTuningSpec veo_tuning_spec = 33;
   }
 
   // Output only. Identifier. Resource name of a TuningJob. Format:
@@ -577,6 +580,48 @@ message TunedModelRef {
   }
 }
 
+// Hyperparameters for Veo.
+message VeoHyperParameters {
+  // An enum defining the tuning task used for Veo.
+  enum TuningTask {
+    // Default value. This value is unused.
+    TUNING_TASK_UNSPECIFIED = 0;
+
+    // Tuning task for image to video.
+    TUNING_TASK_I2V = 1;
+
+    // Tuning task for text to video.
+    TUNING_TASK_T2V = 2;
+  }
+
+  // Optional. Number of complete passes the model makes over the entire
+  // training dataset during training.
+  int64 epoch_count = 1 [(google.api.field_behavior) = OPTIONAL];
+
+  // Optional. Multiplier for adjusting the default learning rate.
+  double learning_rate_multiplier = 2 [(google.api.field_behavior) = OPTIONAL];
+
+  // Optional. The tuning task. Either I2V or T2V.
+  TuningTask tuning_task = 3 [(google.api.field_behavior) = OPTIONAL];
+}
+
+// Tuning Spec for Veo Model Tuning.
+message VeoTuningSpec {
+  // Required. Training dataset used for tuning. The dataset can be specified as
+  // either a Cloud Storage path to a JSONL file or as the resource name of a
+  // Vertex Multimodal Dataset.
+  string training_dataset_uri = 1 [(google.api.field_behavior) = REQUIRED];
+
+  // Optional. Validation dataset used for tuning. The dataset can be specified
+  // as either a Cloud Storage path to a JSONL file or as the resource name of a
+  // Vertex Multimodal Dataset.
+  string validation_dataset_uri = 2 [(google.api.field_behavior) = OPTIONAL];
+
+  // Optional. Hyperparameters for Veo.
+  VeoHyperParameters hyper_parameters = 3
+      [(google.api.field_behavior) = OPTIONAL];
+}
+
 // TunedModelCheckpoint for the Tuned Model of a Tuning Job.
 message TunedModelCheckpoint {
   // The ID of the checkpoint.
