diff --git a/dlc_developer_config.toml b/dlc_developer_config.toml index 207784d1f191..f5ec442d051e 100644 --- a/dlc_developer_config.toml +++ b/dlc_developer_config.toml @@ -15,7 +15,7 @@ neuronx_mode = false graviton_mode = false # Please only set it to true if you are preparing a ARM64 related PR # Do remember to revert it back to false before merging any PR (including ARM64 dedicated PR) -arm64_mode = false +arm64_mode = true # Please only set it to True if you are preparing a HABANA related PR # Do remember to revert it back to False before merging any PR (including HABANA dedicated PR) habana_mode = false @@ -36,12 +36,12 @@ deep_canary_mode = false [build] # Add in frameworks you would like to build. By default, builds are disabled unless you specify building an image. -# available frameworks - ["base", "vllm", "sglang", "autogluon", "huggingface_vllm", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"] -build_frameworks = [] +# available frameworks - ["base", "vllm", "sglang", "autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"] +build_frameworks = ["pytorch"] # By default we build both training and inference containers. Set true/false values to determine which to build. -build_training = true +build_training = false build_inference = true # Set do_build to "false" to skip builds and test the latest image built by this PR diff --git a/pytorch/inference/buildspec-arm64-2-6-ec2.yml b/pytorch/inference/buildspec-arm64-2-6-ec2.yml index 5b30650f5745..5829fb656594 100644 --- a/pytorch/inference/buildspec-arm64-2-6-ec2.yml +++ b/pytorch/inference/buildspec-arm64-2-6-ec2.yml @@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch version: &VERSION 2.6.0 short_version: &SHORT_VERSION "2.6" arch_type: arm64 -# autopatch_build: "True" +autopatch_build: "True" repository_info: inference_repository: &INFERENCE_REPOSITORY diff --git a/pytorch/inference/buildspec-arm64-2-6-sm.yml b/pytorch/inference/buildspec-arm64-2-6-sm.yml index 6a4bf68c35e0..e7df86655c62 100644 --- a/pytorch/inference/buildspec-arm64-2-6-sm.yml +++ b/pytorch/inference/buildspec-arm64-2-6-sm.yml @@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch version: &VERSION 2.6.0 short_version: &SHORT_VERSION "2.6" arch_type: arm64 -# autopatch_build: "True" +autopatch_build: "True" repository_info: inference_repository: &INFERENCE_REPOSITORY