Skip to content

Commit b9633de

Browse files
test tensorflow 2.19 EC2 x86
1 parent 8c8f8ac commit b9633de

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

dlc_developer_config.toml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ neuronx_mode = false
1515
graviton_mode = false
1616
# Please only set it to true if you are preparing a ARM64 related PR
1717
# Do remember to revert it back to false before merging any PR (including ARM64 dedicated PR)
18-
arm64_mode = true
18+
arm64_mode = false
1919
# Please only set it to True if you are preparing a HABANA related PR
2020
# Do remember to revert it back to False before merging any PR (including HABANA dedicated PR)
2121
habana_mode = false
@@ -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", "autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"]
40-
build_frameworks = ["pytorch"]
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.
@@ -122,8 +122,8 @@ use_scheduler = false
122122
dlc-pr-base = ""
123123

124124
# Standard Framework Training
125-
dlc-pr-pytorch-training = "pytorch/training/buildspec-arm64-2-7-ec2.yml"
126-
dlc-pr-tensorflow-2-training = ""
125+
dlc-pr-pytorch-training = ""
126+
dlc-pr-tensorflow-2-training = "tensorflow/training/buildspec-2-19-sm.yml"
127127
dlc-pr-autogluon-training = ""
128128

129129
# ARM64 Training

0 commit comments

Comments
 (0)