chore(aiplatform); Gen AI code samples update to suport the recent product changes (GoogleCloudPlatform#8452)

gericdong · web-flow · commit 5e7853fdfb04 · 2023-07-20T11:25:53.000-04:00
diff --git a/aiplatform/src/main/java/aiplatform/CreatePipelineJobModelTuningSample.java b/aiplatform/src/main/java/aiplatform/CreatePipelineJobModelTuningSample.java
@@ -33,13 +33,12 @@ public class CreatePipelineJobModelTuningSample {
   public static void main(String[] args) throws IOException {
     // TODO(developer): Replace these variables before running the sample.
     String project = "PROJECT";
-    String location =
-        "europe-west4"; // Model tuning is only supported in europe-west4 for Public Preview
+    String location = "europe-west4"; // europe-west4 and us-central1 are the supported regions
     String pipelineJobDisplayName = "PIPELINE_JOB_DISPLAY_NAME";
     String modelDisplayName = "MODEL_DISPLAY_NAME";
     String outputDir = "OUTPUT_DIR";
     String datasetUri = "DATASET_URI";
-    int trainingSteps = 100;
+    int trainingSteps = 300;
 
     createPipelineJobModelTuningSample(
         project,
@@ -88,7 +87,7 @@ public static void createPipelineJobModelTuningSample(
       PipelineJob pipelineJob =
           PipelineJob.newBuilder()
               .setTemplateUri(
-                  "https://us-kfp.pkg.dev/ml-pipeline/large-language-model-pipelines/tune-large-model/v1.0.0")
+                  "https://us-kfp.pkg.dev/ml-pipeline/large-language-model-pipelines/tune-large-model/v2.0.0")
               .setDisplayName(pipelineJobDisplayName)
               .setRuntimeConfig(runtimeConfig)
               .build();
diff --git a/aiplatform/src/test/java/aiplatform/CreatePipelineJobModelTuningSampleTest.java b/aiplatform/src/test/java/aiplatform/CreatePipelineJobModelTuningSampleTest.java
@@ -51,7 +51,7 @@ public class CreatePipelineJobModelTuningSampleTest {
       "gs://ucaip-samples-europe-west4/training_pipeline_output";
   private static final String DATASET_URI =
       "gs://cloud-samples-data/ai-platform/generative_ai/headline_classification.jsonl";
-  private static final int TRAINING_STEPS = 100;
+  private static final int TRAINING_STEPS = 300;
   private String pipelineJobName;
   private ByteArrayOutputStream bout;
   private PrintStream originalPrintStream;
