Skip to content

Commit 93f6513

Browse files
tushar00jaingithubsgi
authored andcommitted
add script to train with ft (pytorch#1812)
Summary: the script adds configuration options to run training locally with ft enabled --- [//]: # (BEGIN SAPLING FOOTER) Stack created with [Sapling](https://sapling-scm.com). Best reviewed with [ReviewStack](https://reviewstack.dev/pytorch/torchtitan/pull/1812). * pytorch#1840 * pytorch#1811 * pytorch#1810 * __->__ pytorch#1812 * pytorch#1809 --------- Co-authored-by: Tushar Jain <[email protected]>
1 parent d209acc commit 93f6513

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

torchtitan/models/llama3_ft/train_configs/debug_model.toml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,13 @@
11
[job]
22
dump_folder = "./outputs"
3+
<<<<<<< HEAD
34
description = "Llama 3 fault-tolerant debug training"
45
print_config = false
56
custom_config_module = "torchtitan.components.ft.config"
7+
=======
8+
description = "Llama 3 debug training"
9+
print_args = false
10+
>>>>>>> f63037ff (add script to train with ft (#1812))
611

712
[profiling]
813
enable_profiling = true

0 commit comments

Comments
 (0)