Skip to content

Commit 07ef325

Browse files
author
Bhanu Teja Goshikonda
committed
building image and running basic tests
1 parent 7f874ff commit 07ef325

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

dlc_developer_config.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,12 @@ 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", "autogluon", "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.
4444
build_training = true
45-
build_inference = true
45+
build_inference = false
4646

4747
# Set do_build to "false" to skip builds and test the latest image built by this PR
4848
# Note: at least one build is required to set do_build to "false"
@@ -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-7-sm.yml"
128128
dlc-pr-tensorflow-2-training = ""
129129
dlc-pr-autogluon-training = ""
130130

0 commit comments

Comments
 (0)