Skip to content

Commit 08d6cff

Browse files
authored
fix: Change default cluster for DAG pw_mcjax_benchmark_recipe (#1186)
This change updates the cluster for DAG `pw_mcjax_benchmark_recipe`
1 parent 7060812 commit 08d6cff

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dags/maxtext_pathways/configs/parameters.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343
),
4444
),
4545
"cluster_name": Param(
46-
"pw-v6e-8",
46+
"pw-v6e-32x4",
4747
type="string",
4848
title="Cluster Name",
4949
description="GCP cluster name for training model.",
@@ -68,7 +68,7 @@
6868
enum=DEVICE_VERSION,
6969
),
7070
"core_count": Param(
71-
8,
71+
32,
7272
type="integer",
7373
title="Core Count",
7474
description='Device core count for the cluster. ex: v6e-"64"',

0 commit comments

Comments
 (0)