Skip to content

Commit 6527e23

Browse files
authored
Merge branch 'master' into modernize-metadata
2 parents be72fcf + c200b4f commit 6527e23

File tree

8 files changed

+131
-5
lines changed

8 files changed

+131
-5
lines changed

CHANGELOG.md

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,26 @@
11
# Changelog
22

3+
## v2.227.0 (2024-07-30)
4+
5+
### Features
6+
7+
* added code scanning through CodeQL
8+
9+
### Bug Fixes and Other Changes
10+
11+
* Fixed cpu isntance type for the estimator register test
12+
* update image_uri_configs 07-29-2024 11:28:28 PST
13+
* avoid AccessDenied error for a while on SageMaker Studio wtih do…
14+
* SMP PT 2.3 Fix
15+
* chore: pin framework version in serverless inference tests
16+
* image uri in TGI 2.2.0 image
17+
* explicitly access enum member values to avoid Python version related regression
18+
* chore: add huggingface TGI 2.2.0 config
19+
* update image_uri_configs 07-22-2024 11:53:54 PST
20+
* update image_uri_configs 07-17-2024 07:17:38 PST
21+
* update image_uri_configs 07-16-2024 07:17:45 PST
22+
* add support for new regions
23+
324
## v2.226.1 (2024-07-17)
425

526
## v2.226.0 (2024-07-12)

VERSION

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
2.226.2.dev0
1+
2.227.1.dev0

src/sagemaker/image_uri_config/huggingface-tei-cpu.json

Lines changed: 49 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@
44
"cpu"
55
],
66
"version_aliases": {
7-
"1.2": "1.2.3"
7+
"1.2": "1.2.3",
8+
"1.4": "1.4.0"
89
},
910
"versions": {
1011
"1.2.3": {
@@ -53,6 +54,53 @@
5354
"container_version": {
5455
"cpu": "ubuntu22.04"
5556
}
57+
},
58+
"1.4.0":{
59+
"py_versions": [
60+
"py310"
61+
],
62+
"registries": {
63+
"af-south-1": "510948584623",
64+
"ap-east-1": "651117190479",
65+
"ap-northeast-1": "354813040037",
66+
"ap-northeast-2": "366743142698",
67+
"ap-northeast-3": "867004704886",
68+
"ap-south-1": "720646828776",
69+
"ap-south-2": "628508329040",
70+
"ap-southeast-1": "121021644041",
71+
"ap-southeast-2": "783357654285",
72+
"ap-southeast-3": "951798379941",
73+
"ap-southeast-4": "106583098589",
74+
"ca-central-1": "341280168497",
75+
"ca-west-1": "190319476487",
76+
"cn-north-1": "450853457545",
77+
"cn-northwest-1": "451049120500",
78+
"eu-central-1": "492215442770",
79+
"eu-central-2": "680994064768",
80+
"eu-north-1": "662702820516",
81+
"eu-south-1": "978288397137",
82+
"eu-south-2": "104374241257",
83+
"eu-west-1": "141502667606",
84+
"eu-west-2": "764974769150",
85+
"eu-west-3": "659782779980",
86+
"il-central-1": "898809789911",
87+
"me-central-1": "272398656194",
88+
"me-south-1": "801668240914",
89+
"sa-east-1": "737474898029",
90+
"us-east-1": "683313688378",
91+
"us-east-2": "257758044811",
92+
"us-gov-east-1": "237065988967",
93+
"us-gov-west-1": "414596584902",
94+
"us-iso-east-1": "833128469047",
95+
"us-isob-east-1": "281123927165",
96+
"us-west-1": "746614075791",
97+
"us-west-2": "246618743249"
98+
},
99+
"tag_prefix": "2.0.1-tei1.4.0",
100+
"repository": "tei-cpu",
101+
"container_version": {
102+
"cpu": "ubuntu22.04"
103+
}
56104
}
57105
}
58106
}

src/sagemaker/image_uri_config/huggingface-tei.json

Lines changed: 49 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@
44
"gpu"
55
],
66
"version_aliases": {
7-
"1.2": "1.2.3"
7+
"1.2": "1.2.3",
8+
"1.4": "1.4.0"
89
},
910
"versions": {
1011
"1.2.3": {
@@ -53,6 +54,53 @@
5354
"container_version": {
5455
"gpu": "cu122-ubuntu22.04"
5556
}
57+
},
58+
"1.4.0": {
59+
"py_versions": [
60+
"py310"
61+
],
62+
"registries": {
63+
"af-south-1": "510948584623",
64+
"ap-east-1": "651117190479",
65+
"ap-northeast-1": "354813040037",
66+
"ap-northeast-2": "366743142698",
67+
"ap-northeast-3": "867004704886",
68+
"ap-south-1": "720646828776",
69+
"ap-south-2": "628508329040",
70+
"ap-southeast-1": "121021644041",
71+
"ap-southeast-2": "783357654285",
72+
"ap-southeast-3": "951798379941",
73+
"ap-southeast-4": "106583098589",
74+
"ca-central-1": "341280168497",
75+
"ca-west-1": "190319476487",
76+
"cn-north-1": "450853457545",
77+
"cn-northwest-1": "451049120500",
78+
"eu-central-1": "492215442770",
79+
"eu-central-2": "680994064768",
80+
"eu-north-1": "662702820516",
81+
"eu-south-1": "978288397137",
82+
"eu-south-2": "104374241257",
83+
"eu-west-1": "141502667606",
84+
"eu-west-2": "764974769150",
85+
"eu-west-3": "659782779980",
86+
"il-central-1": "898809789911",
87+
"me-central-1": "272398656194",
88+
"me-south-1": "801668240914",
89+
"sa-east-1": "737474898029",
90+
"us-east-1": "683313688378",
91+
"us-east-2": "257758044811",
92+
"us-gov-east-1": "237065988967",
93+
"us-gov-west-1": "414596584902",
94+
"us-iso-east-1": "833128469047",
95+
"us-isob-east-1": "281123927165",
96+
"us-west-1": "746614075791",
97+
"us-west-2": "246618743249"
98+
},
99+
"tag_prefix": "2.0.1-tei1.4.0",
100+
"repository": "tei",
101+
"container_version": {
102+
"gpu": "cu122-ubuntu22.04"
103+
}
56104
}
57105
}
58106
}

src/sagemaker/image_uri_config/pytorch.json

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1008,6 +1008,7 @@
10081008
"us-gov-west-1": "442386744353",
10091009
"us-iso-east-1": "886529160074",
10101010
"us-isob-east-1": "094389454867",
1011+
"us-isof-east-1": "303241398832",
10111012
"us-isof-south-1": "454834333376",
10121013
"us-west-1": "763104351884",
10131014
"us-west-2": "763104351884"
@@ -1052,6 +1053,7 @@
10521053
"us-gov-west-1": "442386744353",
10531054
"us-iso-east-1": "886529160074",
10541055
"us-isob-east-1": "094389454867",
1056+
"us-isof-east-1": "303241398832",
10551057
"us-isof-south-1": "454834333376",
10561058
"us-west-1": "763104351884",
10571059
"us-west-2": "763104351884"
@@ -2331,6 +2333,7 @@
23312333
"us-gov-west-1": "442386744353",
23322334
"us-iso-east-1": "886529160074",
23332335
"us-isob-east-1": "094389454867",
2336+
"us-isof-east-1": "303241398832",
23342337
"us-isof-south-1": "454834333376",
23352338
"us-west-1": "763104351884",
23362339
"us-west-2": "763104351884"
@@ -2419,6 +2422,7 @@
24192422
"us-gov-west-1": "442386744353",
24202423
"us-iso-east-1": "886529160074",
24212424
"us-isob-east-1": "094389454867",
2425+
"us-isof-east-1": "303241398832",
24222426
"us-isof-south-1": "454834333376",
24232427
"us-west-1": "763104351884",
24242428
"us-west-2": "763104351884"

src/sagemaker/image_uri_config/tensorflow.json

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2140,6 +2140,7 @@
21402140
"us-gov-west-1": "442386744353",
21412141
"us-iso-east-1": "886529160074",
21422142
"us-isob-east-1": "094389454867",
2143+
"us-isof-east-1": "303241398832",
21432144
"us-isof-south-1": "454834333376",
21442145
"us-west-1": "763104351884",
21452146
"us-west-2": "763104351884"
@@ -2181,6 +2182,7 @@
21812182
"us-gov-west-1": "442386744353",
21822183
"us-iso-east-1": "886529160074",
21832184
"us-isob-east-1": "094389454867",
2185+
"us-isof-east-1": "303241398832",
21842186
"us-isof-south-1": "454834333376",
21852187
"us-west-1": "763104351884",
21862188
"us-west-2": "763104351884"
@@ -4354,6 +4356,7 @@
43544356
"us-gov-west-1": "442386744353",
43554357
"us-iso-east-1": "886529160074",
43564358
"us-isob-east-1": "094389454867",
4359+
"us-isof-east-1": "303241398832",
43574360
"us-isof-south-1": "454834333376",
43584361
"us-west-1": "763104351884",
43594362
"us-west-2": "763104351884"

tests/integ/test_byo_estimator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def test_byo_estimator(sagemaker_session, region, cpu_instance_type, training_se
109109

110110

111111
@pytest.mark.release
112-
def test_estimator_register_publish_training_details(sagemaker_session, region):
112+
def test_estimator_register_publish_training_details(sagemaker_session, region, cpu_instance_type):
113113

114114
bucket = sagemaker_session.default_bucket()
115115
prefix = "model-card-sample-notebook"
@@ -150,7 +150,7 @@ def test_estimator_register_publish_training_details(sagemaker_session, region):
150150
container,
151151
role="SageMakerRole",
152152
instance_count=1,
153-
instance_type="ml.m4.xlarge",
153+
instance_type=cpu_instance_type,
154154
output_path=output_location,
155155
sagemaker_session=sagemaker_session,
156156
)

tests/unit/sagemaker/image_uris/test_huggingface_llm.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,11 @@
2121
TEI_VERSIONS_MAPPING = {
2222
"gpu": {
2323
"1.2.3": "2.0.1-tei1.2.3-gpu-py310-cu122-ubuntu22.04",
24+
"1.4.0": "2.0.1-tei1.4.0-gpu-py310-cu122-ubuntu22.04",
2425
},
2526
"cpu": {
2627
"1.2.3": "2.0.1-tei1.2.3-cpu-py310-ubuntu22.04",
28+
"1.4.0": "2.0.1-tei1.4.0-cpu-py310-ubuntu22.04",
2729
},
2830
}
2931
HF_VERSIONS_MAPPING = {

0 commit comments

Comments
 (0)