Merge branch 'main' into batch

GoogleCloudPlatform · Jul 24, 2023 · fddbedf · fddbedf
2 parents 5e60db3 + 020c773
commit fddbedf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ai-platform/snippets/tuning.js b/ai-platform/snippets/tuning.js
@@ -23,7 +23,7 @@ async function main(
   gcsOutputDirectory,
   location = 'europe-west4',
   datasetUri = 'gs://cloud-samples-data/ai-platform/generative_ai/headline_classification.jsonl',
-  trainSteps = 10
+  trainSteps = 300
 ) {
   // [START aiplatform_model_tuning]
   /**
@@ -66,7 +66,7 @@ async function main(
 
     const pipelineJob = {
       templateUri:
-        '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',
       displayName: 'my-tuning-job',
       runtimeConfig,
     };