Skip to content

Commit 2081516

Browse files
fix(hf): correct HF neuronx pytorch version (#5501)
The pytorch version was incorrect, the available image is v2.8 and not v2.7. Co-authored-by: Molly He <mollyhe@amazon.com>
1 parent 1f8cd0f commit 2081516

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sagemaker-core/src/sagemaker/core/image_uri_config/huggingface-neuronx.json

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -640,9 +640,9 @@
640640
},
641641
"4.55.4": {
642642
"version_aliases": {
643-
"pytorch2.7": "pytorch2.7.1"
643+
"pytorch2.8": "pytorch2.8.0"
644644
},
645-
"pytorch2.7.1": {
645+
"pytorch2.8.0": {
646646
"py_versions": [
647647
"py310"
648648
],

0 commit comments

Comments
 (0)