@@ -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 = 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)
2121habana_mode = false
@@ -37,11 +37,11 @@ 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 = []
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 = true
44+ build_training = false
4545build_inference = true
4646
4747# Set do_build to "false" to skip builds and test the latest image built by this PR
@@ -59,11 +59,11 @@ notify_test_failures = false
5959# ## On by default
6060sanity_tests = true
6161security_tests = true
62- safety_check_test = false
63- ecr_scan_allowlist_feature = false
64- ecs_tests = true
65- eks_tests = true
66- ec2_tests = true
62+ safety_check_test = true
63+ ecr_scan_allowlist_feature = true
64+ ecs_tests = false
65+ eks_tests = false
66+ ec2_tests = false
6767# Set it to true if you are preparing a Benchmark related PR
6868ec2_benchmark_tests = false
6969
@@ -74,7 +74,7 @@ ec2_benchmark_tests = false
7474ec2_tests_on_heavy_instances = false
7575# ## SM specific tests
7676# ## On by default
77- sagemaker_local_tests = true
77+ sagemaker_local_tests = false
7878# ## Set enable_ipv6 = true to run tests with IPv6-enabled resources
7979# ## Off by default (set to false)
8080enable_ipv6 = false
@@ -92,7 +92,7 @@ enable_ipv6 = false
9292ipv6_vpc_name = " "
9393
9494# run standard sagemaker remote tests from test/sagemaker_tests
95- sagemaker_remote_tests = true
95+ sagemaker_remote_tests = false
9696# run efa sagemaker tests
9797sagemaker_efa_tests = false
9898# run release_candidate_integration tests
@@ -153,7 +153,7 @@ dlc-pr-tensorflow-2-habana-training = ""
153153
154154# Standard Framework Inference
155155dlc-pr-pytorch-inference = " "
156- dlc-pr-tensorflow-2-inference = " "
156+ dlc-pr-tensorflow-2-inference = " tensorflow/inference/buildspec-arm64-2-18-sm.yml "
157157dlc-pr-autogluon-inference = " "
158158
159159# Graviton Inference
0 commit comments