diff --git a/pyproject.toml b/pyproject.toml
index 2c9593f..ba2bd6d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,7 +22,7 @@ dependencies = [
     "dynaconf~=3.2.7",
     "torch",
     "botocore",
-    "sagemaker==2.244.2",
+    "sagemaker==3.4.0",
     "numpy",
     "requests",
     "accelerate",
diff --git a/requirements.txt b/requirements.txt
index 0e60ddc..414c436 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -275,7 +275,7 @@ safetensors==0.5.3
     # via
     #   accelerate
     #   transformers
-sagemaker==2.236.0
+sagemaker==3.4.0
     # via wraval (pyproject.toml)
 sagemaker-core==1.0.16
     # via sagemaker