Skip to content

Commit 091385d

Browse files
authored
change: HuggingFace PT Neuronx release in SDK (#4237)
1 parent d355d5b commit 091385d

File tree

2 files changed

+46
-3
lines changed

2 files changed

+46
-3
lines changed

src/sagemaker/image_uri_config/huggingface-neuronx.json

Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,49 @@
9494
"container_version": {"inf": "ubuntu20.04"},
9595
"sdk_versions": ["sdk2.9.1"]
9696
}
97+
},
98+
"4.34.1": {
99+
"version_aliases": {"pytorch1.13": "pytorch1.13.1"},
100+
"pytorch1.13.1": {
101+
"py_versions": ["py310"],
102+
"repository": "huggingface-pytorch-inference-neuronx",
103+
"registries": {
104+
"af-south-1": "626614931356",
105+
"il-central-1": "780543022126",
106+
"ap-east-1": "871362719292",
107+
"ap-northeast-1": "763104351884",
108+
"ap-northeast-2": "763104351884",
109+
"ap-northeast-3": "364406365360",
110+
"ap-south-1": "763104351884",
111+
"ap-south-2": "772153158452",
112+
"ap-southeast-1": "763104351884",
113+
"ap-southeast-2": "763104351884",
114+
"ap-southeast-4": "457447274322",
115+
"ca-central-1": "763104351884",
116+
"cn-north-1": "727897471807",
117+
"cn-northwest-1": "727897471807",
118+
"eu-central-1": "763104351884",
119+
"eu-central-2": "380420809688",
120+
"eu-north-1": "763104351884",
121+
"eu-west-1": "763104351884",
122+
"eu-west-2": "763104351884",
123+
"eu-west-3": "763104351884",
124+
"eu-south-1": "692866216735",
125+
"eu-south-2": "503227376785",
126+
"me-south-1": "217643126080",
127+
"sa-east-1": "763104351884",
128+
"us-east-1": "763104351884",
129+
"us-east-2": "763104351884",
130+
"us-gov-east-1": "446045086412",
131+
"us-gov-west-1": "442386744353",
132+
"us-iso-east-1": "886529160074",
133+
"us-isob-east-1": "094389454867",
134+
"us-west-1": "763104351884",
135+
"us-west-2": "763104351884"
136+
},
137+
"container_version": {"inf": "ubuntu20.04"},
138+
"sdk_versions": ["sdk2.15.0"]
139+
}
97140
}
98141
}
99142
}

tests/conftest.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -441,12 +441,12 @@ def huggingface_neuron_latest_inference_transformer_version():
441441

442442
@pytest.fixture(scope="module")
443443
def huggingface_neuronx_latest_inference_transformer_version():
444-
return "4.28"
444+
return "4.34.1"
445445

446446

447447
@pytest.fixture(scope="module")
448448
def huggingface_neuronx_latest_training_transformer_version():
449-
return "4.28"
449+
return "4.34.1"
450450

451451

452452
@pytest.fixture(scope="module")
@@ -456,7 +456,7 @@ def huggingface_neuron_latest_inference_py_version():
456456

457457
@pytest.fixture(scope="module")
458458
def huggingface_neuronx_latest_inference_py_version():
459-
return "py38"
459+
return "py310"
460460

461461

462462
@pytest.fixture(scope="module")

0 commit comments

Comments
 (0)