diff --git a/sagemaker-train/tests/data/train/script_mode/requirements.txt b/sagemaker-train/tests/data/train/script_mode/requirements.txt index f7b8ccf0cc..ada0b275b2 100644 --- a/sagemaker-train/tests/data/train/script_mode/requirements.txt +++ b/sagemaker-train/tests/data/train/script_mode/requirements.txt @@ -1,3 +1,3 @@ numpy -f https://download.pytorch.org/whl/torch_stable.html -torch==2.7.0 +torch==2.8.0