Skip to content

Commit ca86599

Browse files
author
Depp Lee
committed
Fix base.yml path
1 parent 0cf6e23 commit ca86599

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dags/orbax/util/test_config_util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -189,7 +189,7 @@ def generate_workload_command(
189189
command = (
190190
f"export TPU_PREMAPPED_BUFFER_SIZE={tpu_premmapped_size} && "
191191
f"export TPU_PREMAPPED_BUFFER_TRANSFER_THRESHOLD_BYTES={tpu_premmapped_size} && "
192-
"python3 -m MaxText.train MaxText/configs/base.yml "
192+
"python3 -m MaxText.train maxtext/configs/base.yml "
193193
"remat_policy=full "
194194
"global_parameter_scale=1 "
195195
f"base_output_directory={posixpath.join(self.base_dir, out_folder)} "

0 commit comments

Comments
 (0)