Skip to content

Commit ca9fd66

Browse files
author
Bhanu Teja Goshikonda
committed
Configure build for PyTorch 2.10 training EC2 images
1 parent 8dbd698 commit ca9fd66

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dlc_developer_config.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ deep_canary_mode = false
3737
[build]
3838
# Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image.
3939
# available frameworks - ["base", "vllm", "sglang", "autogluon", "huggingface_vllm", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"]
40-
build_frameworks = []
40+
build_frameworks = ["pytorch"]
4141

4242

4343
# By default we build both training and inference containers. Set true/false values to determine which to build.
@@ -124,7 +124,7 @@ nightly_pr_test_mode = false
124124
dlc-pr-base = ""
125125

126126
# Standard Framework Training
127-
dlc-pr-pytorch-training = ""
127+
dlc-pr-pytorch-training = "pytorch/training/buildspec-2-10-ec2.yml"
128128
dlc-pr-tensorflow-2-training = ""
129129
dlc-pr-autogluon-training = ""
130130

0 commit comments

Comments
 (0)