Skip to content

Commit 1b9cd89

Browse files
chore: Enable TF 2.19 training builds in developer config
1 parent aeac886 commit 1b9cd89

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", "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 = ["tensorflow"]
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"
@@ -125,7 +125,7 @@ dlc-pr-base = ""
125125

126126
# Standard Framework Training
127127
dlc-pr-pytorch-training = ""
128-
dlc-pr-tensorflow-2-training = ""
128+
dlc-pr-tensorflow-2-training = "tensorflow/training/buildspec-2-19-sm.yml"
129129
dlc-pr-autogluon-training = ""
130130

131131
# ARM64 Training

0 commit comments

Comments
 (0)