Skip to content

Commit 2830f39

Browse files
evakravievakravi
authored
Merge branch 'master' into fix/jumpstart-estimator-gated-uncompressed-training
2 parents c9ceff2 + 681d211 commit 2830f39

File tree

16 files changed

+661
-75
lines changed

16 files changed

+661
-75
lines changed

‎CHANGELOG.md

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

3+
## v2.244.1 (2025-05-15)
4+
5+
### Bug Fixes and Other Changes
6+
7+
* Fix Flask-Limiter version
8+
* Fix test_huggingface_tei_uris()
9+
* huggingface-llm-neuronx dlc
10+
* huggingface-neuronx dlc image_uri
11+
* huggingface-tei dlc image_uri
12+
* Fix test_deploy_with_update_endpoint()
13+
* add AG v1.3
14+
* parameter mismatch in update_endpoint
15+
* remove --strip-component for untar source tar.gz
16+
* Fix type annotations
17+
* chore: Allow omegaconf >=2.2,<3
18+
* honor json serialization of HPs
19+
* Map llama models to correct script
20+
* pin test dependency
21+
* fix bad initialization script error message
22+
* Improve error logging and documentation for issue 4007
23+
* build(deps): bump scikit-learn
24+
* build(deps): bump mlflow
25+
* build(deps): bump mlflow in /tests/data/serve_resources/mlflow/pytorch
26+
* chore: Add tei 1.6.0 image
27+
328
## v2.244.0 (2025-05-02)
429

530
### Features

‎VERSION

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

‎pyproject.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,8 @@ dependencies = [
5555
"tblib>=1.7.0,<4",
5656
"tqdm",
5757
"urllib3>=1.26.8,<3.0.0",
58-
"uvicorn"
58+
"uvicorn",
59+
"graphene>=3,<4"
5960
]
6061

6162
[project.scripts]

‎requirements/extras/test_requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ stopit==1.1.2
1616
# Update tox.ini to have correct version of airflow constraints file
1717
apache-airflow==2.10.4
1818
apache-airflow-providers-amazon==7.2.1
19-
Flask-Limiter==3.12
19+
Flask-Limiter==3.11
2020
attrs>=23.1.0,<24
2121
fabric==3.2.2
2222
requests==2.32.2

‎src/sagemaker/image_uri_config/huggingface-llm-neuronx.json

Lines changed: 54 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
"inf2"
55
],
66
"version_aliases": {
7-
"0.0": "0.0.27"
7+
"0.0": "0.0.28"
88
},
99
"versions": {
1010
"0.0.16": {
@@ -589,6 +589,59 @@
589589
"container_version": {
590590
"inf2": "ubuntu22.04"
591591
}
592+
},
593+
"0.0.28": {
594+
"py_versions": [
595+
"py310"
596+
],
597+
"registries": {
598+
"af-south-1": "626614931356",
599+
"ap-east-1": "871362719292",
600+
"ap-east-2": "975050140332",
601+
"ap-northeast-1": "763104351884",
602+
"ap-northeast-2": "763104351884",
603+
"ap-northeast-3": "364406365360",
604+
"ap-south-1": "763104351884",
605+
"ap-south-2": "772153158452",
606+
"ap-southeast-1": "763104351884",
607+
"ap-southeast-2": "763104351884",
608+
"ap-southeast-3": "907027046896",
609+
"ap-southeast-4": "457447274322",
610+
"ap-southeast-5": "550225433462",
611+
"ap-southeast-7": "590183813437",
612+
"ca-central-1": "763104351884",
613+
"ca-west-1": "204538143572",
614+
"cn-north-1": "727897471807",
615+
"cn-northwest-1": "727897471807",
616+
"eu-central-1": "763104351884",
617+
"eu-central-2": "380420809688",
618+
"eu-north-1": "763104351884",
619+
"eu-south-1": "692866216735",
620+
"eu-south-2": "503227376785",
621+
"eu-west-1": "763104351884",
622+
"eu-west-2": "763104351884",
623+
"eu-west-3": "763104351884",
624+
"il-central-1": "780543022126",
625+
"me-central-1": "914824155844",
626+
"me-south-1": "217643126080",
627+
"mx-central-1": "637423239942",
628+
"sa-east-1": "763104351884",
629+
"us-east-1": "763104351884",
630+
"us-east-2": "763104351884",
631+
"us-gov-east-1": "446045086412",
632+
"us-gov-west-1": "442386744353",
633+
"us-iso-east-1": "886529160074",
634+
"us-isob-east-1": "094389454867",
635+
"us-isof-east-1": "303241398832",
636+
"us-isof-south-1": "454834333376",
637+
"us-west-1": "763104351884",
638+
"us-west-2": "763104351884"
639+
},
640+
"tag_prefix": "2.1.2-optimum0.0.28",
641+
"repository": "huggingface-pytorch-tgi-inference",
642+
"container_version": {
643+
"inf2": "ubuntu22.04"
644+
}
592645
}
593646
}
594647
}

‎src/sagemaker/image_uri_config/huggingface-neuron.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
],
1818
"repository": "huggingface-pytorch-inference-neuron",
1919
"registries": {
20+
"ap-east-2": "975050140332",
2021
"ap-northeast-1": "763104351884",
2122
"ap-south-1": "763104351884",
2223
"ap-south-2": "772153158452",

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

Lines changed: 141 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,9 @@
66
"version_aliases": {
77
"4.28": "4.28.1",
88
"4.34": "4.34.1",
9-
"4.36": "4.36.2"
9+
"4.36": "4.36.2",
10+
"4.43": "4.43.2",
11+
"4.48": "4.48.1"
1012
},
1113
"versions": {
1214
"4.28.1": {
@@ -19,6 +21,7 @@
1921
],
2022
"repository": "huggingface-pytorch-training-neuronx",
2123
"registries": {
24+
"ap-east-2": "975050140332",
2225
"ap-northeast-1": "763104351884",
2326
"ap-south-1": "763104351884",
2427
"ap-south-2": "772153158452",
@@ -62,6 +65,7 @@
6265
],
6366
"repository": "huggingface-pytorch-inference-neuronx",
6467
"registries": {
68+
"ap-east-2": "975050140332",
6569
"ap-northeast-1": "763104351884",
6670
"ap-south-1": "763104351884",
6771
"ap-south-2": "772153158452",
@@ -105,6 +109,7 @@
105109
],
106110
"repository": "huggingface-pytorch-inference-neuronx",
107111
"registries": {
112+
"ap-east-2": "975050140332",
108113
"ap-northeast-1": "763104351884",
109114
"ap-south-1": "763104351884",
110115
"ap-south-2": "772153158452",
@@ -137,6 +142,92 @@
137142
"sdk2.18.0"
138143
]
139144
}
145+
},
146+
"4.43.2": {
147+
"version_aliases": {
148+
"pytorch2.1": "pytorch2.1.2"
149+
},
150+
"pytorch2.1.2": {
151+
"py_versions": [
152+
"py310"
153+
],
154+
"repository": "huggingface-pytorch-inference-neuronx",
155+
"registries": {
156+
"ap-northeast-1": "763104351884",
157+
"ap-south-1": "763104351884",
158+
"ap-south-2": "772153158452",
159+
"ap-southeast-1": "763104351884",
160+
"ap-southeast-2": "763104351884",
161+
"ap-southeast-4": "457447274322",
162+
"ap-southeast-5": "550225433462",
163+
"ap-southeast-7": "590183813437",
164+
"cn-north-1": "727897471807",
165+
"cn-northwest-1": "727897471807",
166+
"eu-central-1": "763104351884",
167+
"eu-central-2": "380420809688",
168+
"eu-south-2": "503227376785",
169+
"eu-west-1": "763104351884",
170+
"eu-west-3": "763104351884",
171+
"il-central-1": "780543022126",
172+
"mx-central-1":"637423239942",
173+
"sa-east-1": "763104351884",
174+
"us-east-1": "763104351884",
175+
"us-east-2": "763104351884",
176+
"us-gov-east-1": "446045086412",
177+
"us-gov-west-1": "442386744353",
178+
"us-west-2": "763104351884",
179+
"ca-west-1": "204538143572"
180+
},
181+
"container_version": {
182+
"inf": "ubuntu20.04"
183+
},
184+
"sdk_versions": [
185+
"sdk2.20.0"
186+
]
187+
}
188+
},
189+
"4.48.1": {
190+
"version_aliases": {
191+
"pytorch2.1": "pytorch2.1.2"
192+
},
193+
"pytorch2.1.2": {
194+
"py_versions": [
195+
"py310"
196+
],
197+
"repository": "huggingface-pytorch-inference-neuronx",
198+
"registries": {
199+
"ap-northeast-1": "763104351884",
200+
"ap-south-1": "763104351884",
201+
"ap-south-2": "772153158452",
202+
"ap-southeast-1": "763104351884",
203+
"ap-southeast-2": "763104351884",
204+
"ap-southeast-4": "457447274322",
205+
"ap-southeast-5": "550225433462",
206+
"ap-southeast-7": "590183813437",
207+
"cn-north-1": "727897471807",
208+
"cn-northwest-1": "727897471807",
209+
"eu-central-1": "763104351884",
210+
"eu-central-2": "380420809688",
211+
"eu-south-2": "503227376785",
212+
"eu-west-1": "763104351884",
213+
"eu-west-3": "763104351884",
214+
"il-central-1": "780543022126",
215+
"mx-central-1":"637423239942",
216+
"sa-east-1": "763104351884",
217+
"us-east-1": "763104351884",
218+
"us-east-2": "763104351884",
219+
"us-gov-east-1": "446045086412",
220+
"us-gov-west-1": "442386744353",
221+
"us-west-2": "763104351884",
222+
"ca-west-1": "204538143572"
223+
},
224+
"container_version": {
225+
"inf": "ubuntu20.04"
226+
},
227+
"sdk_versions": [
228+
"sdk2.20.0"
229+
]
230+
}
140231
}
141232
}
142233
},
@@ -147,7 +238,8 @@
147238
"version_aliases": {
148239
"4.28": "4.28.1",
149240
"4.34": "4.34.1",
150-
"4.36": "4.36.2"
241+
"4.36": "4.36.2",
242+
"4.43": "4.43.2"
151243
},
152244
"versions": {
153245
"4.28.1": {
@@ -163,6 +255,7 @@
163255
"af-south-1": "626614931356",
164256
"il-central-1": "780543022126",
165257
"ap-east-1": "871362719292",
258+
"ap-east-2": "975050140332",
166259
"ap-northeast-1": "763104351884",
167260
"ap-northeast-2": "763104351884",
168261
"ap-northeast-3": "364406365360",
@@ -218,6 +311,7 @@
218311
"af-south-1": "626614931356",
219312
"il-central-1": "780543022126",
220313
"ap-east-1": "871362719292",
314+
"ap-east-2": "975050140332",
221315
"ap-northeast-1": "763104351884",
222316
"ap-northeast-2": "763104351884",
223317
"ap-northeast-3": "364406365360",
@@ -274,6 +368,7 @@
274368
"af-south-1": "626614931356",
275369
"il-central-1": "780543022126",
276370
"ap-east-1": "871362719292",
371+
"ap-east-2": "975050140332",
277372
"ap-northeast-1": "763104351884",
278373
"ap-northeast-2": "763104351884",
279374
"ap-northeast-3": "364406365360",
@@ -324,6 +419,7 @@
324419
"af-south-1": "626614931356",
325420
"il-central-1": "780543022126",
326421
"ap-east-1": "871362719292",
422+
"ap-east-2": "975050140332",
327423
"ap-northeast-1": "763104351884",
328424
"ap-northeast-2": "763104351884",
329425
"ap-northeast-3": "364406365360",
@@ -365,6 +461,49 @@
365461
"sdk2.18.0"
366462
]
367463
}
464+
},
465+
"4.43.2": {
466+
"version_aliases": {
467+
"pytorch2.1": "pytorch2.1.2"
468+
},
469+
"pytorch2.1.2": {
470+
"py_versions": [
471+
"py310"
472+
],
473+
"repository": "huggingface-pytorch-inference-neuronx",
474+
"registries": {
475+
"ap-northeast-1": "763104351884",
476+
"ap-south-1": "763104351884",
477+
"ap-south-2": "772153158452",
478+
"ap-southeast-1": "763104351884",
479+
"ap-southeast-2": "763104351884",
480+
"ap-southeast-4": "457447274322",
481+
"ap-southeast-5": "550225433462",
482+
"ap-southeast-7": "590183813437",
483+
"cn-north-1": "727897471807",
484+
"cn-northwest-1": "727897471807",
485+
"eu-central-1": "763104351884",
486+
"eu-central-2": "380420809688",
487+
"eu-south-2": "503227376785",
488+
"eu-west-1": "763104351884",
489+
"eu-west-3": "763104351884",
490+
"il-central-1": "780543022126",
491+
"mx-central-1":"637423239942",
492+
"sa-east-1": "763104351884",
493+
"us-east-1": "763104351884",
494+
"us-east-2": "763104351884",
495+
"us-gov-east-1": "446045086412",
496+
"us-gov-west-1": "442386744353",
497+
"us-west-2": "763104351884",
498+
"ca-west-1": "204538143572"
499+
},
500+
"container_version": {
501+
"inf": "ubuntu20.04"
502+
},
503+
"sdk_versions": [
504+
"sdk2.20.0"
505+
]
506+
}
368507
}
369508
}
370509
}

0 commit comments

Comments
 (0)