revert toml file changes and pt 2.8 sm changes

Author: DevakiBolleneni

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 = true
+arm64_mode = false
 # 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
@@ -37,12 +37,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_tensorflow", "huggingface_pytorch", "huggingface_tensorflow_trcomp", "huggingface_pytorch_trcomp", "pytorch_trcomp", "tensorflow", "pytorch", "stabilityai_pytorch"]
-build_frameworks = ["pytorch"]
+build_frameworks = []
 
 
 # By default we build both training and inference containers. Set true/false values to determine which to build.
 build_training = true
-build_inference = false
+build_inference = true
 
 # Set do_build to "false" to skip builds and test the latest image built by this PR
 # Note: at least one build is required to set do_build to "false"
@@ -129,7 +129,7 @@ dlc-pr-tensorflow-2-training = ""
 dlc-pr-autogluon-training = ""
 
 # ARM64 Training
-dlc-pr-pytorch-arm64-training = "pytorch/training/buildspec-arm64-2-7-ec2.yml"
+dlc-pr-pytorch-arm64-training = ""
 
 # HuggingFace Training
 dlc-pr-huggingface-tensorflow-training = ""

pytorch/training/docker/2.8/py3/Dockerfile.cpu

@@ -286,14 +286,12 @@ WORKDIR /
 
 # Install SM packages
 RUN pip install --no-cache-dir -U \
-    # address package regression caused by smclarify dependency s3fs
-    "awscli<=1.42.61" \
-    "boto3<=1.40.61" \
     smclarify \
-    "sagemaker>=2.9.0,<3" \
-    "sagemaker-experiments<1" \
+    "sagemaker==2.254.1" \
+    sagemaker-experiments \
     sagemaker-pytorch-training \
     sagemaker-training \
+    # Add sniffio explicitly as it's not included in sagemaker==2.254.1 dependencies
     sniffio
 
 # Install extra packages

pytorch/training/docker/2.8/py3/cu129/Dockerfile.gpu

@@ -251,15 +251,16 @@ WORKDIR /
 
 # Install SM packages
 RUN pip install --no-cache-dir -U \
-    # address package regression caused by smclarify dependency s3fs
-    "awscli<=1.42.61" \
-    "boto3<=1.40.61" \
+    # address package regression caused by smclarify depedency s3fs"
+    "awscli<1.42.50" \
+    "boto3<1.40.50" \
     smclarify \
-    "sagemaker>=2.9.0,<3" \
-    "sagemaker-experiments<1" \
+    "sagemaker==2.254.1" \
+    sagemaker-experiments \
     sagemaker-pytorch-training \
     sagemaker-training \
-    sniffio
+    # Add sniffio explicitly as it's not included in sagemaker==2.254.1 dependencies
+    sniffio   
 
 # Install extra packages
 RUN pip install --no-cache-dir -U \