diff --git a/huggingface/pytorch/tei/docker/tei-requirements.txt b/huggingface/pytorch/tei/docker/tei-requirements.txt
index 55504e8..63cefe0 100644
--- a/huggingface/pytorch/tei/docker/tei-requirements.txt
+++ b/huggingface/pytorch/tei/docker/tei-requirements.txt
@@ -2,9 +2,9 @@ boto3
 dataclasses
 docker
 gitpython
-sagemaker
+sagemaker<3.0.0
 
 parameterized
 pytest
 pytest-mock
-pytest-xdist
\ No newline at end of file
+pytest-xdist
diff --git a/huggingface/pytorch/tgi/docker/tgi-requirements.txt b/huggingface/pytorch/tgi/docker/tgi-requirements.txt
index 55504e8..63cefe0 100644
--- a/huggingface/pytorch/tgi/docker/tgi-requirements.txt
+++ b/huggingface/pytorch/tgi/docker/tgi-requirements.txt
@@ -2,9 +2,9 @@ boto3
 dataclasses
 docker
 gitpython
-sagemaker
+sagemaker<3.0.0
 
 parameterized
 pytest
 pytest-mock
-pytest-xdist
\ No newline at end of file
+pytest-xdist
diff --git a/huggingface/pytorch/tgillamacpp/docker/tgi-llamacpp-requirements.txt b/huggingface/pytorch/tgillamacpp/docker/tgi-llamacpp-requirements.txt
index 55504e8..63cefe0 100644
--- a/huggingface/pytorch/tgillamacpp/docker/tgi-llamacpp-requirements.txt
+++ b/huggingface/pytorch/tgillamacpp/docker/tgi-llamacpp-requirements.txt
@@ -2,9 +2,9 @@ boto3
 dataclasses
 docker
 gitpython
-sagemaker
+sagemaker<3.0.0
 
 parameterized
 pytest
 pytest-mock
-pytest-xdist
\ No newline at end of file
+pytest-xdist