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 = {