retest pt sm local tests

Author: zhuofuAMZ

dlc_developer_config.toml

@@ -37,7 +37,7 @@ 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 - ["autogluon", "huggingface_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"]
-build_frameworks = ["tensorflow"]
+build_frameworks = ["pytorch"]
 
 
 # By default we build both training and inference containers. Set true/false values to determine which to build.
@@ -105,7 +105,7 @@ use_scheduler = false
 
 # Standard Framework Training
 dlc-pr-pytorch-training = ""
-dlc-pr-tensorflow-2-training = "tensorflow/training/buildspec-2-18-sm.yml"
+dlc-pr-tensorflow-2-training = "pytorch/training/buildspec-2-6-sm.yml"
 dlc-pr-autogluon-training = ""
 
 # ARM64 Training
@@ -135,7 +135,7 @@ dlc-pr-tensorflow-2-habana-training = ""
 
 # Standard Framework Inference
 dlc-pr-pytorch-inference = ""
-dlc-pr-tensorflow-2-inference = "tensorflow/inference/buildspec-2-18-sm.yml"
+dlc-pr-tensorflow-2-inference = "pytorch/inference/buildspec-2-6-sm.yml"
 dlc-pr-autogluon-inference = ""
 
 # Graviton Inference