diff --git a/sagemaker-core/src/sagemaker/core/image_uri_config/huggingface-neuronx.json b/sagemaker-core/src/sagemaker/core/image_uri_config/huggingface-neuronx.json index c13e0d4a89..6840eab17e 100644 --- a/sagemaker-core/src/sagemaker/core/image_uri_config/huggingface-neuronx.json +++ b/sagemaker-core/src/sagemaker/core/image_uri_config/huggingface-neuronx.json @@ -640,9 +640,9 @@ }, "4.55.4": { "version_aliases": { - "pytorch2.7": "pytorch2.7.1" + "pytorch2.8": "pytorch2.8.0" }, - "pytorch2.7.1": { + "pytorch2.8.0": { "py_versions": [ "py310" ],