Skip to content

Commit 2c008f8

Browse files
authored
feat: add new optional accelerator_type parameter (#3576)
1 parent 5701105 commit 2c008f8

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ai-platform/snippets/tuning.js

+1
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ async function main(
5757
dataset_uri: helpers.toValue(datasetUri),
5858
large_model_reference: helpers.toValue(model),
5959
model_display_name: helpers.toValue(modelDisplayName),
60+
accelerator_type: helpers.toValue('GPU'), // Optional: GPU or TPU
6061
};
6162

6263
const runtimeConfig = {

0 commit comments

Comments
 (0)