From 2c008f81febaacb568eebfcc0f158c9215f8139b Mon Sep 17 00:00:00 2001
From: Eric Dong <itseric@google.com>
Date: Mon, 20 Nov 2023 11:59:27 -0500
Subject: [PATCH] feat: add new optional accelerator_type parameter (#3576)

---
 ai-platform/snippets/tuning.js | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ai-platform/snippets/tuning.js b/ai-platform/snippets/tuning.js
index 98baa41d18..372159f4c0 100644
--- 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 = {