Skip to content

Commit 5964b25

Browse files
committed
Update torchtitan CI
1 parent 2928160 commit 5964b25

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.github/workflows/test_torchtitan.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,8 @@ jobs:
4848
4949
# Run TorchTitan training with AutoParallel
5050
NGPU=4 CONFIG_FILE="./torchtitan/models/llama3/train_configs/debug_model.toml" ./run_train.sh \
51-
--model.name auto_parallel.llama3 \
51+
--model.name autoparallel.llama3 \
5252
--parallelism.tensor_parallel_degree 4 \
5353
--training.dataset c4 \
5454
--compile.enable
55+
--job.custom_config_module=torchtitan.experiments.autoparallel.job_config

0 commit comments

Comments
 (0)