Skip to content

Commit 3b91e7f

Browse files
jeffnvidiaTaekyungHeo
authored andcommitted
use official online container
1 parent 65abb93 commit 3b91e7f

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

conf/v0.6/general/test_template/nemo_launcher.toml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,15 +3,15 @@ name = "NeMoLauncher"
33
[cmd_args]
44
[cmd_args.repository_url]
55
type = "str"
6-
default = "NEMO_REPOSITORY_URL"
6+
default = "https://github.com/NVIDIA/NeMo-Framework-Launcher.git"
77

88
[cmd_args.repository_commit_hash]
99
type = "str"
10-
default = "6528780fba8185bf61e7c2396fdd2331ee5933a1"
10+
default = "cf411a9ede3b466677df8ee672bcc6c396e71e1a"
1111

1212
[cmd_args.docker_image_url]
1313
type = "str"
14-
default = "DOCKER_IMAGE_URL"
14+
default = "nvcr.io/nvidian/nemofw-training:24.01.01"
1515

1616
[cmd_args.stages]
1717
type = "str"
@@ -33,8 +33,8 @@ name = "NeMoLauncher"
3333
default = "8"
3434

3535
[cmd_args.training]
36-
values = ["gpt3/43b"]
37-
default = "gpt3/43b"
36+
values = ["gpt3/40b_improved"]
37+
default = "gpt3/40b_improved"
3838
[cmd_args.training.exp_manager]
3939
[cmd_args.training.exp_manager.create_checkpoint_callback]
4040
type = "bool"

0 commit comments

Comments
 (0)