Skip to content

Commit 3a90ccf

Browse files
JingyaHuangJingyaHuang
authored
[Feature - Hugging Face] Update Transformers 4.26 - PyTorch 1.13.1 Image uri (#3670)
1 parent 5f75d34 commit 3a90ccf

File tree

2 files changed

+108
-11
lines changed

2 files changed

+108
-11
lines changed

‎src/sagemaker/image_uri_config/huggingface.json

Lines changed: 90 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,8 @@
88
"4.10": "4.10.2",
99
"4.11": "4.11.0",
1010
"4.12": "4.12.3",
11-
"4.17": "4.17.0"
11+
"4.17": "4.17.0",
12+
"4.26": "4.26.0"
1213
},
1314
"versions": {
1415
"4.4.2": {
@@ -772,6 +773,47 @@
772773
"repository": "huggingface-tensorflow-training",
773774
"container_version": {"gpu": "cu112-ubuntu20.04"}
774775
}
776+
},
777+
"4.26.0": {
778+
"version_aliases": {
779+
"pytorch1.13": "pytorch1.13.1"
780+
},
781+
"pytorch1.13.1": {
782+
"py_versions": ["py39"],
783+
"registries": {
784+
"af-south-1": "626614931356",
785+
"ap-east-1": "871362719292",
786+
"ap-northeast-1": "763104351884",
787+
"ap-northeast-2": "763104351884",
788+
"ap-northeast-3": "364406365360",
789+
"ap-south-1": "763104351884",
790+
"ap-southeast-1": "763104351884",
791+
"ap-southeast-2": "763104351884",
792+
"ap-southeast-3": "907027046896",
793+
"ca-central-1": "763104351884",
794+
"cn-north-1": "727897471807",
795+
"cn-northwest-1": "727897471807",
796+
"eu-central-1": "763104351884",
797+
"eu-north-1": "763104351884",
798+
"eu-west-1": "763104351884",
799+
"eu-west-2": "763104351884",
800+
"eu-west-3": "763104351884",
801+
"eu-south-1": "692866216735",
802+
"me-south-1": "217643126080",
803+
"me-central-1": "914824155844",
804+
"sa-east-1": "763104351884",
805+
"us-east-1": "763104351884",
806+
"us-east-2": "763104351884",
807+
"us-gov-east-1": "446045086412",
808+
"us-gov-west-1": "442386744353",
809+
"us-iso-east-1": "886529160074",
810+
"us-isob-east-1": "094389454867",
811+
"us-west-1": "763104351884",
812+
"us-west-2": "763104351884"
813+
},
814+
"repository": "huggingface-pytorch-training",
815+
"container_version": {"gpu": "cu117-ubuntu20.04"}
816+
}
775817
}
776818
}
777819
},
@@ -783,7 +825,8 @@
783825
"4.10": "4.10.2",
784826
"4.11": "4.11.0",
785827
"4.12": "4.12.3",
786-
"4.17": "4.17.0"
828+
"4.17": "4.17.0",
829+
"4.26": "4.26.0"
787830
},
788831
"versions": {
789832
"4.6.1": {
@@ -1337,6 +1380,51 @@
13371380
"repository": "huggingface-tensorflow-inference",
13381381
"container_version": {"gpu": "cu112-ubuntu20.04", "cpu": "ubuntu20.04" }
13391382
}
1383+
},
1384+
"4.26.0": {
1385+
"version_aliases": {
1386+
"pytorch1.13": "pytorch1.13.1"
1387+
},
1388+
"pytorch1.13.1": {
1389+
"py_versions": ["py39"],
1390+
"registries": {
1391+
"af-south-1": "626614931356",
1392+
"ap-east-1": "871362719292",
1393+
"ap-northeast-1": "763104351884",
1394+
"ap-northeast-2": "763104351884",
1395+
"ap-northeast-3": "364406365360",
1396+
"ap-south-1": "763104351884",
1397+
"ap-south-2": "772153158452",
1398+
"ap-southeast-1": "763104351884",
1399+
"ap-southeast-2": "763104351884",
1400+
"ap-southeast-3": "907027046896",
1401+
"ap-southeast-4": "457447274322",
1402+
"ca-central-1": "763104351884",
1403+
"cn-north-1": "727897471807",
1404+
"cn-northwest-1": "727897471807",
1405+
"eu-central-1": "763104351884",
1406+
"eu-central-2": "380420809688",
1407+
"eu-north-1": "763104351884",
1408+
"eu-west-1": "763104351884",
1409+
"eu-west-2": "763104351884",
1410+
"eu-west-3": "763104351884",
1411+
"eu-south-1": "692866216735",
1412+
"eu-south-2": "503227376785",
1413+
"me-south-1": "217643126080",
1414+
"me-central-1": "914824155844",
1415+
"sa-east-1": "763104351884",
1416+
"us-east-1": "763104351884",
1417+
"us-east-2": "763104351884",
1418+
"us-gov-east-1": "446045086412",
1419+
"us-gov-west-1": "442386744353",
1420+
"us-iso-east-1": "886529160074",
1421+
"us-isob-east-1": "094389454867",
1422+
"us-west-1": "763104351884",
1423+
"us-west-2": "763104351884"
1424+
},
1425+
"repository": "huggingface-pytorch-inference",
1426+
"container_version": {"gpu": "cu117-ubuntu20.04", "cpu": "ubuntu20.04" }
1427+
}
13401428
}
13411429
}
13421430
}

‎tests/conftest.py

Lines changed: 18 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,12 @@ def huggingface_pytorch_training_version(huggingface_training_version):
273273

274274
@pytest.fixture(scope="module")
275275
def huggingface_pytorch_training_py_version(huggingface_pytorch_training_version):
276-
return "py38" if Version(huggingface_pytorch_training_version) >= Version("1.9") else "py36"
276+
if Version(huggingface_pytorch_training_version) >= Version("1.13"):
277+
return "py39"
278+
elif Version(huggingface_pytorch_training_version) >= Version("1.9"):
279+
return "py38"
280+
else:
281+
return "py36"
277282

278283

279284
@pytest.fixture(scope="module")
@@ -328,9 +333,12 @@ def huggingface_training_compiler_pytorch_py_version(
328333
def huggingface_pytorch_latest_training_py_version(
329334
huggingface_training_pytorch_latest_version,
330335
):
331-
return (
332-
"py38" if Version(huggingface_training_pytorch_latest_version) >= Version("1.9") else "py36"
333-
)
336+
if Version(huggingface_training_pytorch_latest_version) >= Version("1.13"):
337+
return "py39"
338+
elif Version(huggingface_training_pytorch_latest_version) >= Version("1.9"):
339+
return "py38"
340+
else:
341+
return "py36"
334342

335343

336344
@pytest.fixture(scope="module")
@@ -347,11 +355,12 @@ def pytorch_training_compiler_py_version(
347355
def huggingface_pytorch_latest_inference_py_version(
348356
huggingface_inference_pytorch_latest_version,
349357
):
350-
return (
351-
"py38"
352-
if Version(huggingface_inference_pytorch_latest_version) >= Version("1.9")
353-
else "py36"
354-
)
358+
if Version(huggingface_inference_pytorch_latest_version) >= Version("1.13"):
359+
return "py39"
360+
elif Version(huggingface_inference_pytorch_latest_version) >= Version("1.9"):
361+
return "py38"
362+
else:
363+
return "py36"
355364

356365

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

0 commit comments

Comments
 (0)