feat: add new optional accelerator_type parameter (#3576)

GoogleCloudPlatform · Nov 20, 2023 · 2c008f8 · 2c008f8
1 parent 5701105
commit 2c008f8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ai-platform/snippets/tuning.js b/ai-platform/snippets/tuning.js
@@ -57,6 +57,7 @@ async function main(
       dataset_uri: helpers.toValue(datasetUri),
       large_model_reference: helpers.toValue(model),
       model_display_name: helpers.toValue(modelDisplayName),
+      accelerator_type: helpers.toValue('GPU'), // Optional: GPU or TPU
     };
 
     const runtimeConfig = {