Skip to content

Commit 0483fda

Browse files
committed
onboard 2.6 inf to autopatch - ec2
1 parent cce532f commit 0483fda

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
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 = false
18+
arm64_mode = true
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
@@ -36,12 +36,12 @@ deep_canary_mode = false
3636

3737
[build]
3838
# Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image.
39-
# 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 = []
39+
# available frameworks - ["base", "vllm", "sglang", "autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"]
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.
44-
build_training = true
44+
build_training = false
4545
build_inference = true
4646

4747
# Set do_build to "false" to skip builds and test the latest image built by this PR

pytorch/inference/buildspec-arm64-2-6-ec2.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch
55
version: &VERSION 2.6.0
66
short_version: &SHORT_VERSION "2.6"
77
arch_type: arm64
8-
# autopatch_build: "True"
8+
autopatch_build: "True"
99

1010
repository_info:
1111
inference_repository: &INFERENCE_REPOSITORY
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
buildspec_pointer: buildspec-arm64-2-6-sm.yml
1+
buildspec_pointer: buildspec-arm64-2-6-ec2.yml

0 commit comments

Comments
 (0)