diff --git a/pyproject.toml b/pyproject.toml index e90bc15..e9a3c28 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -24,7 +24,7 @@ dependencies = [ "dynaconf~=3.2.7", "torch", "botocore", - "sagemaker==2.244.2", + "sagemaker==2.237.3", "numpy", "requests", "accelerate", diff --git a/requirements.txt b/requirements.txt index 0e60ddc..1f9af2b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -275,7 +275,7 @@ safetensors==0.5.3 # via # accelerate # transformers -sagemaker==2.236.0 +sagemaker==2.237.3 # via wraval (pyproject.toml) sagemaker-core==1.0.16 # via sagemaker