@@ -15,7 +15,7 @@ neuronx_mode = false
1515graviton_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)
2121habana_mode = false
@@ -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 = [" 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.
44- build_training = false
45- build_inference = true
44+ build_training = 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"
@@ -123,7 +123,7 @@ dlc-pr-base = ""
123123
124124# Standard Framework Training
125125dlc-pr-pytorch-training = " "
126- dlc-pr-tensorflow-2-training = " "
126+ dlc-pr-tensorflow-2-training = " tensorflow/training/buildspec-2-19-sm.yml "
127127dlc-pr-autogluon-training = " "
128128
129129# ARM64 Training
@@ -152,7 +152,7 @@ dlc-pr-tensorflow-2-habana-training = ""
152152# ## INFERENCE PR JOBS ###
153153
154154# Standard Framework Inference
155- dlc-pr-pytorch-inference = " pytorch/inference/buildspec-arm64-2-6-sm.yml "
155+ dlc-pr-pytorch-inference = " "
156156dlc-pr-tensorflow-2-inference = " "
157157dlc-pr-autogluon-inference = " "
158158
0 commit comments