Skip to content

Commit c22454b

Browse files
tensorflow arm64 2.19 inference
1 parent 255ffab commit c22454b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dlc_developer_config.toml

Lines changed: 2 additions & 2 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
@@ -123,7 +123,7 @@ dlc-pr-base = ""
123123

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

129129
# ARM64 Training

0 commit comments

Comments
 (0)